removed strict eval

rogeriobonatti · Jun 11, 2022 · 53aef8d · 53aef8d
1 parent f560064
commit 53aef8d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mushr_rhc_ros/src/rhcnode_network_pcl_new.py b/mushr_rhc_ros/src/rhcnode_network_pcl_new.py
@@ -133,7 +133,7 @@ def __init__(self, dtype, params, logger, name):
         if len(saved_model_path_action)>3: # some small number, path must have more
             checkpoint = torch.load(saved_model_path_action, map_location=device)
             # old code for loading model
-            model.load_state_dict(checkpoint['state_dict'])
+            model.load_state_dict(checkpoint['state_dict'], strict=False)
             # new code for loading mode
             # new_checkpoint = OrderedDict()
             # for key in checkpoint['state_dict'].keys():