diff --git a/turtorial/finetune_unet.md b/turtorial/finetune_unet.md index e24535501d9c1ee25b255a8241418b85e78c4571..3bbaa3374c3e0adf0cf9acdaffc20ad93e45b976 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: