mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-09 09:04:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
JinZr 2023-09-15 11:02:20 +08:00
parent bd053ca90a
commit c3dc8f005b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
icefall/checkpoint.py
View File

@ -76,7 +76,7 @@ def save_checkpoint(
if isinstance(model, DDP): if isinstance(model, DDP):
model = model.module model = model.module
checkpoint = { checkpoint = {
"model": model.state_dict(), "model": model.state_dict(),
"optimizer": optimizer.state_dict() if optimizer is not None else None, "optimizer": optimizer.state_dict() if optimizer is not None else None,