Spaces:

aharley
/

alltracker_demo

Sleeping

aharley commited on Jun 22

Commit

b09e850

verified ·

1 Parent(s): 4ae0355

Upload app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -53,7 +53,7 @@ url = "https://huggingface.co/aharley/alltracker/resolve/main/alltracker.pth"
 state_dict = torch.hub.load_state_dict_from_url(url, map_location='cpu')
 model = Net(16)
 count_parameters(model)
-model.load_state_dict(state_dict, strict=True)
 print('loaded ckpt')
 model.cuda()
 for n, p in model.named_parameters():

 state_dict = torch.hub.load_state_dict_from_url(url, map_location='cpu')
 model = Net(16)
 count_parameters(model)
+model.load_state_dict(state_dict['model'], strict=True)
 print('loaded ckpt')
 model.cuda()
 for n, p in model.named_parameters():