diff --git a/medkit/training/trainer.py b/medkit/training/trainer.py index 7e9d93af..76aa7fa1 100644 --- a/medkit/training/trainer.py +++ b/medkit/training/trainer.py @@ -312,8 +312,8 @@ def save(self, epoch: int) -> str: str Path of the checkpoint saved """ - current_date = datetime.datetime.now().strftime("%d-%m-%Y_%H:%M") - name = f"checkpoint_{epoch:03d}_{current_date}" + timestamp = datetime.datetime.utcnow().strftime("%Y%m%dT%H%M%SZ") + name = f"checkpoint_{epoch:03d}_{timestamp}" checkpoint_dir = Path(self.output_dir) / name self.callback.on_save(checkpoint_dir=str(checkpoint_dir))