未验证 提交 60570823 编写于 作者: W wuzewu 提交者: GitHub

Update finetune_unet.md

上级 602a2a8f
...@@ -64,9 +64,6 @@ DATASET: ...@@ -64,9 +64,6 @@ DATASET:
MODEL: MODEL:
MODEL_NAME: "unet" MODEL_NAME: "unet"
DEFAULT_NORM_TYPE: "bn" DEFAULT_NORM_TYPE: "bn"
TRAIN:
PRETRAINED_MODEL_DIR: "./pretrained_model/unet_bn_coco/"
# 其他配置 # 其他配置
TRAIN_CROP_SIZE: (512, 512) TRAIN_CROP_SIZE: (512, 512)
...@@ -76,6 +73,7 @@ AUG: ...@@ -76,6 +73,7 @@ AUG:
FIX_RESIZE_SIZE: (512, 512) FIX_RESIZE_SIZE: (512, 512)
BATCH_SIZE: 4 BATCH_SIZE: 4
TRAIN: TRAIN:
PRETRAINED_MODEL_DIR: "./pretrained_model/unet_bn_coco/"
MODEL_SAVE_DIR: "./saved_model/unet_pet/" MODEL_SAVE_DIR: "./saved_model/unet_pet/"
SNAPSHOT_EPOCH: 10 SNAPSHOT_EPOCH: 10
TEST: TEST:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册