from local

This commit is contained in:
dohe0342 2023-05-25 21:07:31 +09:00
parent f4c33cf7ea
commit 0962ff596d
2 changed files with 1 additions and 1 deletions

View File

@ -1112,7 +1112,7 @@ def train_one_epoch(
if params.print_diagnostics and batch_idx == 5: if params.print_diagnostics and batch_idx == 5:
return return
#print(lora_modules[20].lora.lora_A.weight) print(lora_modules[20].lora.lora_A.weight)
''' '''
if ( if (
rank == 0 rank == 0