Skip to content

GitLab

Commit c9674890 authored by acivgin1's avatar acivgin1
Browse files

version parsing fix

parent 4bdd8dda
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
pcdet/models/detectors/detector3d_template.py
View file @ c9674890
...@@ -330,6 +330,7 @@ class Detector3DTemplate(nn.Module): ...@@ -330,6 +330,7 @@ class Detector3DTemplate(nn.Module):
def _load_state_dict(self, model_state_disk, version, *, strict=True): def _load_state_dict(self, model_state_disk, version, *, strict=True):
state_dict = self.state_dict() # local cache of state_dict state_dict = self.state_dict() # local cache of state_dict
version = version.split("+")[1]
spconv_keys = find_all_spconv_keys(self) spconv_keys = find_all_spconv_keys(self)
update_model_state = {} update_model_state = {}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment