diff --git a/icefall/checkpoint.py b/icefall/checkpoint.py index f7712628b..ea50b32a9 100644 --- a/icefall/checkpoint.py +++ b/icefall/checkpoint.py @@ -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,