Fix typo (#32)
parent
e0104fd65b
commit
ab6e6eef8b
|
@ -331,7 +331,7 @@ def training(rank, conf, output_dir, args):
|
|||
|
||||
results = None # fix bug with it saving
|
||||
|
||||
lr_scheduler = get_lr_scheduler(optimizer=optimizer, conf=conf.train.lr_scheduler)
|
||||
lr_scheduler = get_lr_scheduler(optimizer=optimizer, conf=conf.train.lr_schedule)
|
||||
if args.restore:
|
||||
optimizer.load_state_dict(init_cp["optimizer"])
|
||||
if "lr_scheduler" in init_cp:
|
||||
|
|
Loading…
Reference in New Issue