WireDataset

models/wirenet/wirepoint_rcnn.py

@@ -633,7 +633,7 @@ if __name__ == '__main__':
             with torch.no_grad():
                 for imgs, targets in data_loader_val:
                     print(111)
-                    pred = model(move_to_device(imgs, device), move_to_device(targets, device))
+                    pred = model(move_to_device(imgs, device))
                     print(f"pred:{pred}")
 
 # imgs, targets = next(iter(data_loader))