ensure shape

GreenWizard2015 · Apr 10, 2024 · 1680f9d · 1680f9d
1 parent 8590866
commit 1680f9d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/NN/Nerf2D.py b/NN/Nerf2D.py
@@ -181,7 +181,8 @@ def getChunk(ind, sz):
       latents = self._withExtraLatents(latents=latents, src=src, points=posC)
       # get residuals if needed
       residual = self._withResidual(src, points=posCB)
-      residual = tf.reshape(residual, (flatB, -1))
+      residual = tf.reshape(residual, (-1, 3))
+      tf.assert_equal(tf.shape(residual), (flatB, 3))
       return dict(
         latents=latents, pos=posC, reverseArgs=reverseArgs, value=value,
         residual=residual