From f8c1d95a2590a61bd0f5e15c93017e56dbf6aefe Mon Sep 17 00:00:00 2001 From: Dobromir Popov Date: Sun, 2 Feb 2025 00:57:12 +0200 Subject: [PATCH] renames --- crypto/brian/index.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/crypto/brian/index.py b/crypto/brian/index.py index 98d8990..805d892 100644 --- a/crypto/brian/index.py +++ b/crypto/brian/index.py @@ -75,7 +75,8 @@ def get_best_models(directory): def save_checkpoint(model, epoch, reward, last_dir=LAST_DIR, best_dir=BEST_DIR): """Save the model state always to the last_dir and conditionally to best_dir if reward is high enough.""" timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") - last_filename = f"model_last_epoch_{epoch}_{timestamp}.pt" + # last_filename = f"model_last_epoch_{epoch}_{timestamp}.pt" + last_filename = f"model_last_epoch_{epoch}.pt" last_path = os.path.join(last_dir, last_filename) torch.save({ "epoch": epoch,