We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e1d9969 commit 6861ffbCopy full SHA for 6861ffb
1 file changed
fmpose3d/lib/hrnet/lib/models/pose_hrnet.py
@@ -493,7 +493,7 @@ def init_weights(self, pretrained=''):
493
need_init_state_dict = {}
494
for name, m in pretrained_state_dict.items():
495
if name.split('.')[0] in self.pretrained_layers \
496
- or self.pretrained_layers[0] is '*':
+ or self.pretrained_layers[0] == '*':
497
need_init_state_dict[name] = m
498
self.load_state_dict(need_init_state_dict, strict=False)
499
elif pretrained:
0 commit comments