[fix] failed to save checkpoints in VideoLearner.fit() (#616)

xueeinstein · web-flow · commit 19a875921a72 · 2020-08-31T15:18:46.000-04:00
diff --git 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),
                         ),
                     )

Original file line number	Diff line number	Diff line change
`@@ -273,7 +273,7 @@ def fit(`
`273`	`273`	`self.save(`
`274`	`274`	`os.path.join(`
`275`	`275`	`model_dir,`
`276`		`- "{model_name}_{self.epoch}.pt".format(`
	`276`	`+ "{model_name}_{epoch}.pt".format(`
`277`	`277`	`model_name=model_name, epoch=str(e).zfill(3),`
`278`	`278`	`),`
`279`	`279`	`)`