run training by changing just run.py

configs/ae_body_1.py

@@ -6,6 +6,7 @@ config['gpu'] = '1'
 config['name'] = 'body_torch_1'
 config['notes'] = ' ' 
 config['tags'] = ['torch', 'body'] 
+config['script'] = 'scripts/ae_body_train.py'
 
 config['pretrained_model'] = ''
 

run.py

@@ -1,9 +1,13 @@
 import os
-from configs.ae_static import config
+import sys
-
+from configs.ae_body_1 import config
 
 if __name__ == '__main__':
-
+    # import wandb
+    # wandb.login(host='https://wandb.lnrweb.com', key='local-40a577c9eccab708f1aa81027f21a995596a5838')
     name = config['name']
+    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 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 &')

scripts/ae_body_train.py

@@ -2,13 +2,22 @@ import os
 import sys
 sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/..')
 
-from configs.ae_body_1 import config
+import argparse
+import importlib
+
+parser = argparse.ArgumentParser(description='')
+parser.add_argument('--name', type=str)
+parser.add_argument('--config', type=str)
+args = vars(parser.parse_args())
+
+config_module = importlib.import_module(args['config'])
+config = config_module.config
+# from configs.ae_body_1 import config
 os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
 os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
 
 
 import time
-import argparse
 import wandb
 import random
 import numpy as np
@@ -159,12 +168,12 @@ def train():
 
 
 if __name__ == '__main__':
-    init_wandb()
+    # init_wandb()
     # device = "cuda" if torch.cuda.is_available() else "cpu"
 
-    parser = argparse.ArgumentParser(description='')
+    # parser = argparse.ArgumentParser(description='')
-    parser.add_argument('--name', type=str)
+    # parser.add_argument('--name', type=str)
-    args = vars(parser.parse_args())
+    # args = vars(parser.parse_args())
     
     resolution_modifier = ''
     if config['resolution'] != 512: