compare predictions with diff curvs

2023-01-29 17:34:54 +00:00
parent cbfa129999
commit 1c24bc0d0e
2 changed files with 90 additions and 25 deletions
@@ -95,7 +95,7 @@ def dice_dense(prediction, ground_truth, weight_map=None):
    return tf.reduce_mean(dice_score)


-@tf.function
+# @tf.function
 def train_step(batch):
    with tf.GradientTape() as tape:
        pred = model([batch['uv_curvature'], batch['uv_mask']], training=True)