diff --git a/aaa_helpers.txt b/aaa_helpers.txt
index 9de0bd6..009e33d 100644
--- a/aaa_helpers.txt
+++ b/aaa_helpers.txt
@@ -77,6 +77,7 @@ nohup python cvae.py --name="lhunor_abl_skin_var_4" > nohup/ablation/abl_skin_va
 nohup python cvae_curv.py --name="lhunor_cvae_curv" > nohup/ablation/cvae_curv.out 2>&1 &
 nohup python cvae_curv.py --name="lhunor_cvae_curv_2" > nohup/ablation/cvae_curv_2.out 2>&1 &
 
+nohup python cvae_pyramid.py --name="lhunor_iccv_pyr_32_res" > nohup/iccv/pyr/iccv_pyr_32_res.out 2>&1 &
 
 abl_2_pyr_32_64_128
 abl_2_pyr_64_cape_f_03223_SS
diff --git a/cvae_pyramid.py b/cvae_pyramid.py
index 7226078..260a859 100644
--- a/cvae_pyramid.py
+++ b/cvae_pyramid.py
@@ -209,7 +209,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)