diff --git a/logger.py b/logger.py index 81c91e176085639ac0342d15449d6497e862d71c..5f5404671f2f1e3afe3f2a40fd0145e32252f9b5 100644 --- a/logger.py +++ b/logger.py @@ -7,7 +7,7 @@ import datetime from args import args NOW_STRING = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S") -CSV_FILE = f"output_{args.mode}_{args.model}_{args.dataset}_{NOW_STRING}.csv" +CSV_FILE = f"output_{args.mode}_{args.model}_{args.dataset}_{args.optimizer}_{NOW_STRING}.csv" def init_csv():