config update

aaa_helpers.txt

@@ -1,3 +1,5 @@
+nohup python cvae_pyramid.py --name="lhunor_abl_pyr_32_no_curv_r1" > nohup/review/abl_pyr_32_no_curv_r1.out 2>&1 &
+
 one final convolutional layer for smoothing, on the final coordinates
 
 skirts!!!!

configs/config_abl_pyr_32_skirt.py

@@ -2,7 +2,8 @@ config = dict()
 
 config['gpu'] = '6'
 config['name'] = 'pyr_abl_32_skirt_no_curv_r1'
-config['notes'] = 'review_1'
+config['notes'] = 'no curv, 10p data'
+config['tags'] = ['review_1']
 
 # DON'T FORGET TO RESET SSIM LOSS USAGE
 # DON'T FORGET TO CHECK MODEL LOADING

cvae_pyramid.py

@@ -210,7 +210,7 @@ def plot_3d(mesh_gt, mesh_pred_unposed_levels, faces, mesh_posed, collision, col
     return fig
 
 
-# @tf.function
+@tf.function
 def train_step(batch):
     levels_input, conds = group_inputs_for_levels(batch, config)
 
@@ -733,7 +733,7 @@ def speed_step(levels_input, conds):
 
 
 if __name__ == '__main__':
-    # init_wandb()
+    init_wandb()
     parser = argparse.ArgumentParser(description='')
     parser.add_argument('--name', type=str, default='lhunor')
     args = vars(parser.parse_args())