modified for site test

2024-07-18 17:52:12 +08:00
parent f90ef72cbf
commit e986ec060b
39 changed files with 2279 additions and 375 deletions
--- a/tracking/trackers/reid/pycache/config.cpython-39.pyc
+++ b/tracking/trackers/reid/pycache/config.cpython-39.pyc
--- a/tracking/trackers/reid/pycache/reid_interface.cpython-39.pyc
+++ b/tracking/trackers/reid/pycache/reid_interface.cpython-39.pyc
--- a/tracking/trackers/reid/reid_interface.py
+++ b/tracking/trackers/reid/reid_interface.py
@ -45,8 +45,7 @@ class ReIDInterface:
            ])
        
        
-        self.model = nn.DataParallel(model).to(self.device)
-        
+        # self.model = nn.DataParallel(model).to(self.device)
        self.model = model
        self.model.load_state_dict(torch.load(self.model_path, map_location=self.device))
        self.model.eval()