Spaces:
Sleeping
Sleeping
updated hrnet
Browse files- 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]
|
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:
|