run training by changing just run.py

This commit is contained in:
2023-04-07 22:23:15 +02:00
parent 345475a74f
commit f4d7f72398
3 changed files with 24 additions and 10 deletions
+8 -4
View File
@@ -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 &')