diff --git a/pytracking/utils/params.py b/pytracking/utils/params.py index 15ec1c0..b56fc24 100644 --- a/pytracking/utils/params.py +++ b/pytracking/utils/params.py @@ -45,7 +45,7 @@ class TrackerParams: net = DiMPTorchScriptWrapper( model_dir='exported_weights', device='cuda', - backbone_sd='backbone_state_dict.pt', + backbone_sd='backbone', classifier_sd='classifier_state_dict.pt', bbregressor_sd='bb_regressor_state_dict.pt' )