Skip to content

Commit 6861ffb

Browse files
committed
fix SyntaxWarning in hrnet lib: replace "is" -> "=="
1 parent e1d9969 commit 6861ffb

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

fmpose3d/lib/hrnet/lib/models/pose_hrnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -493,7 +493,7 @@ def init_weights(self, pretrained=''):
493493
need_init_state_dict = {}
494494
for name, m in pretrained_state_dict.items():
495495
if name.split('.')[0] in self.pretrained_layers \
496-
or self.pretrained_layers[0] is '*':
496+
or self.pretrained_layers[0] == '*':
497497
need_init_state_dict[name] = m
498498
self.load_state_dict(need_init_state_dict, strict=False)
499499
elif pretrained:

0 commit comments

Comments
 (0)