This commit is contained in:
Susan Xueqing Liu 2023-03-13 23:19:58 -04:00 коммит произвёл GitHub
Родитель 27b2712016
Коммит a3e770eac5
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 3 удалений

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

@ -732,10 +732,8 @@ class TransformersEstimator(BaseEstimator):
def _delete_one_ckpt(self, ckpt_location):
if self._use_ray is False:
try:
if os.path.exists(ckpt_location):
shutil.rmtree(ckpt_location)
except FileNotFoundError:
logger.warning("checkpoint {} not found".format(ckpt_location))
def cleanup(self):
super().cleanup()