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

minor fix

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

4
icefall/checkpoint.py
View File

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