fix

2023-03-02 19:33:12 +00:00
parent ec51e5e7cb
commit 523e810f36
2 changed files with 2 additions and 1 deletions
@@ -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)