from local

This commit is contained in:
dohe0342 2023-05-24 13:08:08 +09:00
parent 5a148c26b5
commit 810d178208
3 changed files with 2 additions and 1 deletions

Binary file not shown.

View File

@ -1589,7 +1589,8 @@ def run_adapter(rank, world_size, args, wb=None):
if world_size > 1:
logging.info("Using DDP")
model = DDP(model, device_ids=[rank], find_unused_parameters=True)
print('-'*30)
for module in model.modules():
print(module)