diff --git a/configs/pspnet.yaml b/configs/pspnet.yaml index 57886b5ba03ea1c795d0c81b7fb982b708e416c9..fdc960d6af81bcba30128e972f260da05b33b0e8 100644 --- a/configs/pspnet.yaml +++ b/configs/pspnet.yaml @@ -32,7 +32,7 @@ TEST: TEST_MODEL: "snapshots/cityscapes_pspnet50/final" TRAIN: MODEL_SAVE_DIR: "snapshots/cityscapes_pspnet50/" - PRETRAINED_MODEL_DIR: u"pretrained_model/pspnet50_cityscapes/" + PRETRAINED_MODEL_DIR: u"pretrained_model/pspnet50_bn_cityscapes/" SNAPSHOT_EPOCH: 10 SOLVER: LR: 0.001 diff --git a/turtorial/finetune_pspnet.md b/turtorial/finetune_pspnet.md index 34d20dda85cf56399710d0f70d610fed9997a042..931c3c5f7515e2ebec3d4fccf3069ecc6d6c00fb 100644 --- a/turtorial/finetune_pspnet.md +++ b/turtorial/finetune_pspnet.md @@ -76,7 +76,7 @@ AUG: FIX_RESIZE_SIZE: (512, 512) BATCH_SIZE: 4 TRAIN: - PRETRAINED_MODEL_DIR: "./pretrained_model/pspnet50_cityscapes/" + PRETRAINED_MODEL_DIR: "./pretrained_model/pspnet50_bn_cityscapes/" MODEL_SAVE_DIR: "./saved_model/pspnet_pet/" SNAPSHOT_EPOCH: 10 TEST: