ac5113 commited on
Commit
4724540
1 Parent(s): 1169f2a

updated hrnet

Browse files
Files changed (1) hide show
  1. utils/hrnet.py +1 -1
utils/hrnet.py CHANGED
@@ -546,7 +546,7 @@ class PoseHighResolutionNet(nn.Module):
546
  need_init_state_dict = {}
547
  for name, m in pretrained_state_dict.items():
548
  if name.split('.')[0] in self.pretrained_layers \
549
- or self.pretrained_layers[0] is '*':
550
  need_init_state_dict[name] = m
551
  self.load_state_dict(need_init_state_dict, strict=False)
552
  elif pretrained:
 
546
  need_init_state_dict = {}
547
  for name, m in pretrained_state_dict.items():
548
  if name.split('.')[0] in self.pretrained_layers \
549
+ or self.pretrained_layers[0] == '*':
550
  need_init_state_dict[name] = m
551
  self.load_state_dict(need_init_state_dict, strict=False)
552
  elif pretrained: