Update eval_wrapper/eval.py

eval_wrapper/eval.py

@@ -322,13 +322,13 @@ def compute_all_points(pred_dict,batch):
 
 def eval_scene(model, data_dir,visualize=False,rr_addr=None,run_octmae=False,set_conf=5,
                no_input_mask=False,no_pred_mask=False,no_filter_input_view=False,false_positive=None,false_negative=None,n_pred_views=5,
-               do_filter_all_masks=False, dino_model=None,tsdf=False):
+               do_filter_all_masks=False, dino_model=None,tsdf=False, device = 'cpu'):
     
     if dino_model is None:
         # Loading DINOv2 model
         dino_model = torch.hub.load('facebookresearch/dinov2', "dinov2_vitl14_reg")
         dino_model.eval()
-        dino_model.to("cuda")
+        dino_model.to(device)
 
     dataloader_input_view = GenericLoaderSmall(data_dir,n_pred_views=1,pred_input_only=True,false_positive=false_positive,false_negative=false_negative)
     input_view_loader = DataLoader(dataloader_input_view, batch_size=1, shuffle=True, collate_fn=collate)