minor updates

This commit is contained in:
JinZr 2024-09-06 18:16:18 +08:00
parent 91f7b1ce6f
commit 2e5055a847

View File

@ -803,7 +803,6 @@ def run(rank, world_size, args):
model, model,
device_ids=[rank], device_ids=[rank],
find_unused_parameters=True, find_unused_parameters=True,
broadcast_buffers=False,
) )
optimizer_g = torch.optim.AdamW( optimizer_g = torch.optim.AdamW(