Spaces:

LTT
/

DiMeR

Running on Zero

LutaoJiang commited on Apr 30

Commit

c042813

1 Parent(s): 44f9fc8

update

Files changed (1) hide show

models/DiMeR/models/geometry/rep_3d/flexicubes_geometry.py CHANGED Viewed

@@ -61,7 +61,7 @@ class FlexiCubesGeometry(Geometry):
         self.render_type = render_type
         self.ctx = dr.RasterizeCudaContext(device=device)
-        self.verts.requires_grad_(True)
     def getAABB(self):
         return torch.min(self.verts, dim=0).values, torch.max(self.verts, dim=0).values

         self.render_type = render_type
         self.ctx = dr.RasterizeCudaContext(device=device)
+        # self.verts.requires_grad_(True)
     def getAABB(self):
         return torch.min(self.verts, dim=0).values, torch.max(self.verts, dim=0).values