diff --git a/mushr_rhc_ros/src/rhcnode_network_pcl_new.py b/mushr_rhc_ros/src/rhcnode_network_pcl_new.py index 915c0c8..a7d10e1 100755 --- 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():