25 lines
961 B
Python
25 lines
961 B
Python
import os
|
|
import sys
|
|
from configs.direct_3 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
|
|
# wandb.login(key='******************************************')
|
|
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:])
|
|
|
|
print(f'nohup python {script} --name lhunor_{name} --config {config} > nohup/torch/{name}.out 2>&1 &')
|
|
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 &') |