Update hubconf.py

verlab · Feb 11, 2021 · c8cd49a · c8cd49a
1 parent 5d3ff5d
commit c8cd49a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hubconf.py b/hubconf.py
@@ -44,7 +44,7 @@ def SemanticFastForward_RL(pretrained=False, progress=False, sent_emb_size=1024,
     model.vdan.apply(init_weights)  # Apply function "init_weights" to all FC layers of our model.
 
     if pretrained:
-        state_dict = torch.hub.load_state_dict_from_url('https://github.com/verlab/StraightToThePoint_CVPR_2020/releases/download/v1.0.0/sff_rl_vdan_model-5b50b542.pth',
+        state_dict = torch.hub.load_state_dict_from_url('https://github.com/verlab/StraightToThePoint_CVPR_2020/releases/download/v1.0.0/sff_rl_vdan_model-12547244.pth',
                                                         progress=progress)
 
         model.vdan.load_state_dict(state_dict['vdan_state_dict'])