from local

This commit is contained in:
dohe0342 2023-05-25 00:57:51 +09:00
parent d028c22a8a
commit 1fd1f4292b
3 changed files with 1 additions and 1 deletions

View File

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