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), ), )