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,
device_ids=[rank],
find_unused_parameters=True,
broadcast_buffers=False,
)
optimizer_g = torch.optim.AdamW(