runnable on local

ZombaSY · Aug 10, 2022 · addcb2a · addcb2a
1 parent cc97b01
commit addcb2a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 14 deletions.
diff --git a/experiments/pascal/1464/ours/config.yaml b/experiments/pascal/1464/ours/config.yaml
@@ -1,7 +1,7 @@
 dataset: # Required.
   type: pascal_semi
   train:
-    data_root: ../../../../data/VOC2012
+    data_root: /mnt/prj/users/ssy/dataset/pascal
     data_list: ../../../../data/splits/pascal/1464/labeled.txt
     flip: True
     GaussianBlur: False
@@ -11,7 +11,7 @@ dataset: # Required.
       type: rand
       size: [513, 513] # crop image with HxW size
   val:
-    data_root: ../../../../data/VOC2012
+    data_root: /mnt/prj/users/ssy/dataset/pascal
     data_list: ../../../../data/splits/pascal/val.txt
     crop:
       type: center

diff --git a/u2pl/models/resnet.py b/u2pl/models/resnet.py
@@ -365,18 +365,18 @@ def resnet101(pretrained=True, **kwargs):
         pretrained (bool): If True, returns a model pre-trained on ImageNet
     """
     model = ResNet(Bottleneck, [3, 4, 23, 3], **kwargs)
-    if pretrained:
-        model_url = model_urls["resnet101"]
-        state_dict = torch.load(model_url)
-
-        missing_keys, unexpected_keys = model.load_state_dict(state_dict, strict=False)
-        print(
-            f"[Info] Load ImageNet pretrain from '{model_url}'",
-            "\nmissing_keys: ",
-            missing_keys,
-            "\nunexpected_keys: ",
-            unexpected_keys,
-        )
+    # if pretrained:
+    #     model_url = model_urls["resnet101"]
+    #     state_dict = torch.load(model_url)
+    #
+    #     missing_keys, unexpected_keys = model.load_state_dict(state_dict, strict=False)
+    #     print(
+    #         f"[Info] Load ImageNet pretrain from '{model_url}'",
+    #         "\nmissing_keys: ",
+    #         missing_keys,
+    #         "\nunexpected_keys: ",
+    #         unexpected_keys,
+    #     )
     return model