update

yccyenchicheng · Mar 15, 2023 · 7ed2ae9 · 7ed2ae9
1 parent cca3147
commit 7ed2ae9
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/models/sdfusion_img2shape_model.py b/models/sdfusion_img2shape_model.py
@@ -240,15 +240,12 @@ def set_input(self, input=None, gen_order=None, max_sample=None):
         self.img = input['img']
         self.uc_img = torch.zeros_like(self.img).to(self.device)
 
-        self.gt_vox = input['gt_vox']
-
         if max_sample is not None:
             self.x = self.x[:max_sample]
             self.img = self.img[:max_sample]
             self.uc_img = self.uc_img[:max_sample]
-            self.gt_vox = self.gt_vox[:max_sample]
 
-        vars_list = ['x', 'img', 'gt_vox']
+        vars_list = ['x', 'img']
 
         self.tocuda(var_names=vars_list)