mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2023-02-20 10:34:02 +09:00
parent 5384b916fa
commit 273de26183
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/conformer_ctc2/.train.py.swp
View File

Binary file not shown.

BIN
egs/librispeech/ASR/conformer_ctc2/.train_old.py.swp
View File

Binary file not shown.

4
egs/librispeech/ASR/conformer_ctc2/train.py
View File

@ -1060,11 +1060,11 @@ def run(rank, world_size, args):
if checkpoints and checkpoints.get("optimizer") is not None:
logging.info("Loading optimizer state dict")
#optimizer.load_state_dict(checkpoints["optimizer"])
optimizer.load_state_dict(checkpoints["optimizer"])
if checkpoints and checkpoints.get("scheduler") is not None:
logging.info("Loading scheduler state dict")
#scheduler.load_state_dict(checkpoints["scheduler"])
scheduler.load_state_dict(checkpoints["scheduler"])
if params.print_diagnostics:
opts = diagnostics.TensorDiagnosticOptions(