Update eval_wrapper/eval.py

eval_wrapper/eval.py

@@ -39,6 +39,7 @@ class EvalWrapper(torch.nn.Module):
 
     def to(self,device):
         self.model.to(device)
+        return self
 
     def forward(self,x,dino_model=None):
         pred, gt, loss, scale = eval_model(self.model,x,mode='viz',dino_model=dino_model,return_scale=True)