test

app.py

@@ -164,10 +164,10 @@ class DynamicsVisualizer:
         cfg.sim.uniform = True
         cfg.sim.use_pv = False
 
-        # device = torch.device('cuda')
+        device = torch.device('cuda')
 
         self.cfg = cfg
-        # self.device = device
+        self.device = device
         self.k_rel = 8  # knn for relations
         self.k_wgt = 16  # knn for weights
         self.with_bg = True
@@ -180,7 +180,7 @@ class DynamicsVisualizer:
         seed = cfg.seed
         random.seed(seed)
         np.random.seed(seed)
-        # torch.manual_seed(seed)
+        torch.manual_seed(seed)
         # torch.autograd.set_detect_anomaly(True)
         # torch.backends.cudnn.benchmark = True
 
@@ -926,11 +926,6 @@ class DynamicsVisualizer:
     @spaces.GPU
     def reset(self):
         import warp as wp
-        
-        device = torch.device('cuda:0')
-        self.device = device
-        torch.manual_seed(self.cfg.seed)
-        
         wp.init()
         gpus = [int(gpu) for gpu in self.cfg.gpus]
         wp_devices = [wp.get_device(f'cuda:{gpu}') for gpu in gpus]