fix load model when resuming

jeremyfix · Dec 20, 2023 · 2543cad · 2543cad
1 parent f687810
commit 2543cad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/LabsSolutions/02-pytorch-asr/main_ctc.py b/LabsSolutions/02-pytorch-asr/main_ctc.py
@@ -178,7 +178,7 @@ def train(args):
 
     # @SOL
     if args.resume_from is not None:
-        model._load_from_state_dict(torch.load(args.resume_from))
+        model.load_state_dict(torch.load(args.resume_from))
     # SOL@
 
     # Loss, optimizer