From 2c513928cf6dcbce146ca1fcffc1168753c3da2d Mon Sep 17 00:00:00 2001 From: pengmian Date: Thu, 12 Sep 2019 15:01:03 +0800 Subject: [PATCH] add name bn --- configs/pspnet.yaml | 2 +- turtorial/finetune_pspnet.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/configs/pspnet.yaml b/configs/pspnet.yaml index 57886b5b..fdc960d6 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 34d20dda..931c3c5f 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: -- GitLab