config update

2023-01-26 12:17:56 +00:00
parent 44e1d02e5a
commit 19f7e0dfb9
3 changed files with 6 additions and 3 deletions
@@ -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())