diff --git a/main_custom.py b/main_custom.py
index ecbb936ed2eb0522850f276a07d737c115d1a554..ff5a2b19719281784814e51c1c8211275eca6ee4 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -200,6 +200,7 @@ def run(epochs, eval_inter, save_inter, model, data_train, data_val, data_test,
 
 
 def main(args):
+    print('file', args.file)
     if args.wandb is not None:
         os.environ["WANDB_API_KEY"] = 'b4a27ac6b6145e1a5d0ee7f9e2e8c20bd101dccd'
         os.environ["WANDB_MODE"] = "offline"