From 60570823b7beebce450ef2068d93562ffde82a5b Mon Sep 17 00:00:00 2001 From: wuzewu Date: Mon, 9 Sep 2019 20:59:48 +0800 Subject: [PATCH] Update finetune_unet.md --- turtorial/finetune_unet.md | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/turtorial/finetune_unet.md b/turtorial/finetune_unet.md index e2453550..3bbaa337 100644 --- a/turtorial/finetune_unet.md +++ b/turtorial/finetune_unet.md @@ -64,9 +64,6 @@ DATASET: MODEL: MODEL_NAME: "unet" DEFAULT_NORM_TYPE: "bn" -TRAIN: - PRETRAINED_MODEL_DIR: "./pretrained_model/unet_bn_coco/" - # 其他配置 TRAIN_CROP_SIZE: (512, 512) @@ -76,6 +73,7 @@ AUG: FIX_RESIZE_SIZE: (512, 512) BATCH_SIZE: 4 TRAIN: + PRETRAINED_MODEL_DIR: "./pretrained_model/unet_bn_coco/" MODEL_SAVE_DIR: "./saved_model/unet_pet/" SNAPSHOT_EPOCH: 10 TEST: -- GitLab