fix curvature model

2022-09-14 16:53:55 +00:00
parent ac851d2de7
commit 8c0efd10cd
2 changed files with 4 additions and 4 deletions
@@ -74,7 +74,7 @@ class AutoEncoder(tf.keras.Model):

        # x = x * uv_mask[:,:, tf.newaxis]
        # x = x * uv_mask[:,tf.newaxis,:,:]
-        x = x * uv_mask[:,:,:,tf.newaxis]
+        # x = x * uv_mask[:,:,:,tf.newaxis]

        return x