From bd053ca90aa67ddc482709722de0971014d3f635 Mon Sep 17 00:00:00 2001 From: JinZr <60612200+JinZr@users.noreply.github.com> Date: Fri, 15 Sep 2023 11:00:20 +0800 Subject: [PATCH] minor fix --- icefall/checkpoint.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) 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,