From 1beda23ce25210db75b135d375f12587d5b45bfb Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 28 Jan 2023 12:24:17 +0000 Subject: [PATCH] save config files update --- cvae.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cvae.py b/cvae.py index 59d7c75..b66874a 100644 --- a/cvae.py +++ b/cvae.py @@ -45,7 +45,7 @@ from models.cvae_model import CVAENew as CVAE def init_wandb(): - run = wandb.init(project='ablation', name=config['name'], notes=config['notes'], dir='/wandb') + run = wandb.init(project='ablation', name=config['name'], notes=config['notes'], dir='/wandb', tags=config['tags']) artifact = wandb.Artifact('code', type='code') artifact.add_file('cvae.py') artifact.add_file('models/cvae_model.py') @@ -58,7 +58,9 @@ def init_wandb(): artifact.add_file('convnext/models/convnext_tf.py') config_file_names = [key for key in sys.modules.keys() if key.startswith('configs.')] for config_file_name in config_file_names: - artifact.add_file('/'.join(config_file_name.split('.')) + '.py') + file_name = '/'.join(config_file_name.split('.')) + '.py' + if os.path.exists(file_name): + artifact.add_file(file_name) run.log_artifact(artifact) uv2mesh = UV2Mesh(res=256, is_skirt=config['is_skirt'], is_adaptive=False, is_smpl_orig=config['is_cape'])