Spaces:
Build error
Build error
Matteo Sirri
commited on
Commit
·
3e01e59
1
Parent(s):
23918e1
fix: fix typo
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ def load_model(baseline: bool = False):
|
|
22 |
in_features = model.roi_heads.box_predictor.cls_score.in_features
|
23 |
model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)
|
24 |
checkpoint = torch.load(
|
25 |
-
osp.join(os.getcwd(),"
|
26 |
model.load_state_dict(checkpoint["model"])
|
27 |
model.to(device)
|
28 |
model.eval()
|
|
|
22 |
in_features = model.roi_heads.box_predictor.cls_score.in_features
|
23 |
model.roi_heads.box_predictor = FastRCNNPredictor(in_features, 2)
|
24 |
checkpoint = torch.load(
|
25 |
+
osp.join(os.getcwd(), "model_split3_FT_MOT17.pth"), map_location="cpu")
|
26 |
model.load_state_dict(checkpoint["model"])
|
27 |
model.to(device)
|
28 |
model.eval()
|