未验证 提交 0ffe9f34 编写于 作者: Z Zeyu Chen 提交者: GitHub

Update finetune_icnet.md

上级 5af9b3b7
...@@ -77,12 +77,12 @@ AUG: ...@@ -77,12 +77,12 @@ AUG:
FIX_RESIZE_SIZE: (512, 512) FIX_RESIZE_SIZE: (512, 512)
BATCH_SIZE: 4 BATCH_SIZE: 4
TRAIN: TRAIN:
MODEL_SAVE_DIR: "./finetune/icnet_pet/" MODEL_SAVE_DIR: "./saved_model/icnet_pet/"
SNAPSHOT_EPOCH: 10 SNAPSHOT_EPOCH: 10
TEST: TEST:
TEST_MODEL: "./finetune/icnet_pet/final" TEST_MODEL: "./saved_model/icnet_pet/final"
SOLVER: SOLVER:
NUM_EPOCHS: 500 NUM_EPOCHS: 100
LR: 0.005 LR: 0.005
LR_POLICY: "poly" LR_POLICY: "poly"
OPTIMIZER: "adam" OPTIMIZER: "adam"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册