mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
minor updates
This commit is contained in:
parent
91f7b1ce6f
commit
2e5055a847
@ -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(
|
||||||
|
Loading…
x
Reference in New Issue
Block a user