Temp Fix for #38

FluxML · Oct 7, 2020 · 6304fd5 · 6304fd5
1 parent af2bf1e
commit 6304fd5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/rep/mesh.jl b/src/rep/mesh.jl
@@ -870,7 +870,14 @@ end
 
 function _compute_verts_padded(m::TriMesh, refresh::Bool = false)
     if refresh || !(m._verts_padded_valid)
-        _list_to_padded!(m._verts_padded, m._verts_list, 0, (3, m.V))
+
+        # FIXME: #38 chamfer_loss gradient doesn't reach packed, so using temp fix
+        _packed = get_verts_packed(m)
+        _padded = _packed_to_padded(_packed, m._verts_len, 0)
+        m._verts_padded = _padded
+
+        # _list_to_padded!(m._verts_padded, m._verts_list, 0, (3, m.V))
+
         # avoiding setproperty!, as we are building padded
         # from list and list is always valid
         # setfield!(m, :_verts_padded, verts_padded)