mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +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)
|
model.to(device)
|
||||||
if world_size > 1:
|
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])
|
model = DDP(model, device_ids=[rank])
|
||||||
|
|
||||||
optimizer = Noam(
|
optimizer = Noam(
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user