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