From f56053286c1fba219019e1b29454980782a12c34 Mon Sep 17 00:00:00 2001
From: whs <wanghaoshuang@baidu.com>
Date: Fri, 19 Jul 2019 20:21:49 +0800
Subject: [PATCH] Fix doc and train.py of icnet. (#2832)

---
 PaddleCV/icnet/README.md | 4 ++--
 PaddleCV/icnet/train.py  | 8 +++++---
 2 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/PaddleCV/icnet/README.md b/PaddleCV/icnet/README.md
index 4698a41c..4858cae7 100644
--- a/PaddleCV/icnet/README.md
+++ b/PaddleCV/icnet/README.md
@@ -66,7 +66,7 @@ Iter[0]; train loss: 2.338; sub4_loss: 3.367; sub24_loss: 4.120; sub124_loss: 0.
 ### 测试
 执行以下命令在`Cityscape`测试数据集上进行测试：
 ```
-python eval.py --model_path="./cnkpnt/100" --use_gpu=True
+python eval.py --model_path="./chkpnt/100" --use_gpu=True
 ```
 需要通过选项`--model_path`指定模型文件。
 测试脚本的输出的评估指标为[mean IoU]()。
@@ -75,7 +75,7 @@ python eval.py --model_path="./cnkpnt/100" --use_gpu=True
 执行以下命令对指定的数据进行预测：
 ```
 python infer.py \
---model_path="./cnkpnt/100" \
+--model_path="./chkpnt/100" \
 --images_path="./data/cityscape/" \
 --images_list="./data/cityscape/infer.list"
 ```
diff --git a/PaddleCV/icnet/train.py b/PaddleCV/icnet/train.py
index cdb8e96c..3c851479 100644
--- a/PaddleCV/icnet/train.py
+++ b/PaddleCV/icnet/train.py
@@ -35,9 +35,11 @@ LAMBDA2 = 0.4
 LAMBDA3 = 1.0
 LEARNING_RATE = 0.003
 POWER = 0.9
-LOG_PERIOD = 100
-CHECKPOINT_PERIOD = 100
-TOTAL_STEP = 100
+LOG_PERIOD = 1
+CHECKPOINT_PERIOD = 1000
+TOTAL_STEP = 60000
+if 'ce_mode' in os.environ:
+    TOTAL_STEP = 100
 
 no_grad_set = []
 
-- 
GitLab