Skip to content
Snippets Groups Projects

Log when removing bested checkpoint

Merged Yoann Schneider requested to merge log-when-removing-checkpoint into main
1 file
+ 3
0
Compare changes
  • Side-by-side
  • Inline
# -*- coding: utf-8 -*-
import logging
import os
import random
from copy import deepcopy
@@ -28,6 +29,7 @@ from dan.utils import fix_ddp_layers_names, ind_to_token
if MLFLOW_AVAILABLE:
import mlflow
logger = logging.getLogger(__name__)
MODEL_NAMES = ("encoder", "decoder")
@@ -430,6 +432,7 @@ class GenericTrainingManager:
# Remove other checkpoints
for path_to_del in self.paths["checkpoints"].glob(f"{name}_*.pt"):
logger.warning(f"Removing checkpoint `{path_to_del}`")
path_to_del.unlink()
torch.save(content, path)
Loading