From e23b1209a20417628faff06c89d58325c7f860f5 Mon Sep 17 00:00:00 2001 From: Bill Xue Date: Mon, 24 Aug 2020 10:52:43 +0800 Subject: [PATCH] [fix] failed to save checkpoints in VideoLearner.fit() --- utils_cv/action_recognition/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils_cv/action_recognition/model.py b/utils_cv/action_recognition/model.py index 60cf7c90b..afcf9612d 100644 --- a/utils_cv/action_recognition/model.py +++ b/utils_cv/action_recognition/model.py @@ -273,7 +273,7 @@ def fit( self.save( os.path.join( model_dir, - "{model_name}_{self.epoch}.pt".format( + "{model_name}_{epoch}.pt".format( model_name=model_name, epoch=str(e).zfill(3), ), )