Skip to content

Commit

Permalink
Merge pull request ifzhang#13 from cclauss/patch-1
Browse files Browse the repository at this point in the history
Use ==/!= to compare str, bytes, and int literals
  • Loading branch information
ifzhang committed Apr 13, 2020
2 parents 5fa5f07 + 3f780be commit 29e4487
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lib/models/networks/pose_hrnet.py
Expand Up @@ -521,7 +521,7 @@ def init_weights(self, pretrained=''):
need_init_state_dict = {}
for name, m in pretrained_state_dict.items():
if name.split('.')[0] in self.pretrained_layers \
or self.pretrained_layers[0] is '*':
or self.pretrained_layers[0] == '*':
need_init_state_dict[name] = m
self.load_state_dict(need_init_state_dict, strict=False)
elif pretrained:
Expand Down

0 comments on commit 29e4487

Please sign in to comment.