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,