run training by changing just run.py
This commit is contained in:
@@ -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 &')
|
||||
Reference in New Issue
Block a user