add debug logging to cvae_train
This commit is contained in:
@@ -1,22 +1,24 @@
|
||||
import os
|
||||
import sys
|
||||
from configs.ae_curv_1 import config
|
||||
from configs.abl_all import config
|
||||
import argparse
|
||||
|
||||
parser = argparse.ArgumentParser(description='')
|
||||
parser.add_argument('--debug', action='store_true')
|
||||
args = vars(parser.parse_args())
|
||||
# parser = argparse.ArgumentParser(description='')
|
||||
# parser.add_argument('--debug', action='store_true')
|
||||
# args = vars(parser.parse_args())
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
# import wandb
|
||||
# wandb.login(host='https://wandb.lnrweb.com', key='local-40a577c9eccab708f1aa81027f21a995596a5838')
|
||||
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:
|
||||
|
||||
Reference in New Issue
Block a user