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

This commit is contained in:
Bill Xue 2020-09-01 03:18:46 +08:00 коммит произвёл GitHub
Родитель 3b8d2b93f7
Коммит 19a875921a
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

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