Spaces:

WindVChen
/

INR-Harmon

Running

WindVChen commited on Jul 21, 2023

Commit

3ce3e4c

•

1 Parent(s): b1bd18a

Update model/backbone.py

Files changed (1) hide show

model/backbone.py CHANGED Viewed

@@ -67,7 +67,7 @@ class HRNetV2(nn.Module):
         if pretained:
             self.load_pretrained_weights(
-                r".\pretrained_models/hrnetv2_w18_imagenet_pretrained.pth")
         self.output_resolution = (opt.input_size // 8) ** 2

         if pretained:
             self.load_pretrained_weights(
+                "./pretrained_models/hrnetv2_w18_imagenet_pretrained.pth")
         self.output_resolution = (opt.input_size // 8) ** 2