[fix] failed to save checkpoints in VideoLearner.fit() (#616)
This commit is contained in:
Родитель
3b8d2b93f7
Коммит
19a875921a
|
@ -273,7 +273,7 @@ class VideoLearner(object):
|
|||
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),
|
||||
),
|
||||
)
|
||||
|
|
Загрузка…
Ссылка в новой задаче