From 08f26785ea5ad5badace2516e4b5513fd5008ff3 Mon Sep 17 00:00:00 2001 From: Dobromir Popov Date: Sun, 2 Feb 2025 00:58:23 +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 805d892..5199e03 100644 --- a/crypto/brian/index.py +++ b/crypto/brian/index.py @@ -99,7 +99,8 @@ def save_checkpoint(model, epoch, reward, last_dir=LAST_DIR, best_dir=BEST_DIR): # Remove the worst checkpoint. os.remove(os.path.join(best_dir, min_file)) if add_to_best: - best_filename = f"best_{reward:.4f}_epoch_{epoch}_{timestamp}.pt" + # best_filename = f"best_{reward:.4f}_epoch_{epoch}_{timestamp}.pt" + best_filename = f"best_epoch_{epoch}.pt" best_path = os.path.join(best_dir, best_filename) torch.save({ "epoch": epoch,