diff --git a/run.py b/run.py
index 59060cd..ca6fdbc 100644
--- a/run.py
+++ b/run.py
@@ -1,6 +1,13 @@
 import os
 import sys
-from configs.ae_body_1 import config
+from configs.ae_curv_1 import config
+import argparse
+
+parser = argparse.ArgumentParser(description='')
+parser.add_argument('--debug', action='store_true')
+args = vars(parser.parse_args())
+
+
 
 if __name__ == '__main__':
     # import wandb
@@ -9,5 +16,8 @@ if __name__ == '__main__':
     script = config['script']
     config = [key for key in sys.modules.keys()  if key.startswith('configs.')][-1]
     current_file = '/'.join(__file__.split('/')[-2:])
-    # os.system(f'nohup python scripts/ae_static_train.py --name lhunor_{name} > nohup/torch/{name}.out 2>&1 &')
-    os.system(f'nohup python {script} --name lhunor_{name} --config {config} > nohup/torch/{name}.out 2>&1 &')
\ No newline at end of file
+    
+    if args['debug']:
+        os.system(f'python {script} --name lhunor_{name} --config {config}')
+    else:
+        os.system(f'nohup python {script} --name lhunor_{name} --config {config} > nohup/torch/{name}.out 2>&1 &')
\ No newline at end of file