Update predictor.py
Browse files- predictor.py +1 -1
predictor.py
CHANGED
|
@@ -90,7 +90,7 @@ class CoTrackerOnlinePredictor(torch.nn.Module):
|
|
| 90 |
state_dict = torch.load(f, map_location="cpu")
|
| 91 |
if "model" in state_dict:
|
| 92 |
state_dict = state_dict["model"]
|
| 93 |
-
model.
|
| 94 |
print('LOAD STATE DICT')
|
| 95 |
|
| 96 |
|
|
|
|
| 90 |
state_dict = torch.load(f, map_location="cpu")
|
| 91 |
if "model" in state_dict:
|
| 92 |
state_dict = state_dict["model"]
|
| 93 |
+
model.load_state_dict(state_dict)
|
| 94 |
print('LOAD STATE DICT')
|
| 95 |
|
| 96 |
|