Try to fix the gpu aborted problem

zero_shot.py

@@ -121,7 +121,6 @@ class Matcher(object):
         self.fmap_cfg = self.cfg.deepfeat_conf.fmap
         self.dataloaders = dict()
 
-    @spaces.GPU
     def _init(self):
         cfg = self.cfg
         self.fmap_model = DFMNet(self.cfg["deepfeat_conf"]["fmap"]).cuda()
@@ -138,7 +137,6 @@ class Matcher(object):
         self.eye = torch.eye(self.n_fmap).float().cuda()
         self.eye.requires_grad = False
 
-    @spaces.GPU
     def fmap(self, shape_dict, target_dict):
         diff_model_cuda = self.diffusion_model
         diff_model_cuda.net.cuda()
@@ -155,7 +153,6 @@ class Matcher(object):
             mask_12, mask_21 = None, None
         return C12_pred, C12_obj, C21_pred, C21_obj, feat1, feat2, evecs_trans1, evecs_trans2, mask_12, mask_21
 
-    @spaces.GPU
     def zo_shot(self, shape_dict, target_dict):
         self._init()
         evecs1, evecs2 = shape_dict["evecs"], target_dict["evecs"]
@@ -165,7 +162,6 @@ class Matcher(object):
         indKNN_new, _ = extract_p2p_torch_fmap(new_FM, evecs1, evecs2)
         return new_FM, indKNN_new
 
-    @spaces.GPU
     def optimize(self, shape_dict, target_dict, target_normals):
         self._init()
         diff_model_cuda = self.diffusion_model