From c3dc8f005b72eef6f0283e07450d05a222b153a3 Mon Sep 17 00:00:00 2001 From: JinZr <60612200+JinZr@users.noreply.github.com> Date: Fri, 15 Sep 2023 11:02:20 +0800 Subject: [PATCH] minor fix --- icefall/checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/icefall/checkpoint.py b/icefall/checkpoint.py index ea50b32a9..c83c56a53 100644 --- a/icefall/checkpoint.py +++ b/icefall/checkpoint.py @@ -76,7 +76,7 @@ def save_checkpoint( if isinstance(model, DDP): model = model.module - + checkpoint = { "model": model.state_dict(), "optimizer": optimizer.state_dict() if optimizer is not None else None,