mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-09 14:05:33 +00:00
remove note
This commit is contained in:
parent
159803fccb
commit
5802097652
@ -650,8 +650,6 @@ def run(rank, world_size, args):
|
||||
|
||||
model.to(device)
|
||||
if world_size > 1:
|
||||
# Note: find_unused_parameters=True is needed in case we
|
||||
# want to set params.att_rate = 0 (i.e. att decoder is not trained)
|
||||
model = DDP(model, device_ids=[rank])
|
||||
|
||||
optimizer = Noam(
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user