cvae norm training

2023-06-09 08:55:31 +00:00
parent 9147d6daa9
commit 385aea398b
5 changed files with 671 additions and 2 deletions
@@ -51,7 +51,7 @@ from custom_utils.custom_utils import UV2Mesh, Accumulator, plot_3d, get_summary


 def init_wandb():
-    run = wandb.init(project='iccv', name=config['name'], notes=config['notes'], dir='/tmp', tags=config['tags'])
+    run = wandb.init(project='iccv', name=config['name'], notes=config['notes'], dir='/data-local/127/temp', tags=config['tags'])
    artifact = wandb.Artifact('code', type='code')
    artifact.add_file('scripts/cvae.py')
    artifact.add_file('losses/losses.py')