add debug logging to cvae_train

This commit is contained in:
2023-05-21 09:59:57 +00:00
parent 8255c1bbc9
commit fc27a3cedf
2 changed files with 36 additions and 21 deletions
+8 -6
View File
@@ -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: