From f08d894ca414dc5064647ca12bd39d51057b7840 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Thu, 16 Aug 2018 20:48:33 +0800
Subject: [PATCH] Fix ce of icnet.

---
 fluid/icnet/.run_ce.sh |  3 ++-
 fluid/icnet/train.py   | 10 +++++++---
 2 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/fluid/icnet/.run_ce.sh b/fluid/icnet/.run_ce.sh
index a46081c7..643c1ed4 100755
--- a/fluid/icnet/.run_ce.sh
+++ b/fluid/icnet/.run_ce.sh
@@ -2,6 +2,7 @@
 
 # This file is only used for continuous evaluation.
 
+export ce_mode=1
 rm -rf *_factor.txt
-python train.py --use_gpu=True 1> log
+python train.py --use_gpu=True --random_mirror=False --random_scaling=False 1> log
 cat log | python _ce.py
diff --git a/fluid/icnet/train.py b/fluid/icnet/train.py
index b38f0825..9b0f4fd4 100644
--- a/fluid/icnet/train.py
+++ b/fluid/icnet/train.py
@@ -4,6 +4,7 @@ import cityscape
 import argparse
 import functools
 import sys
+import os
 import time
 import paddle.fluid as fluid
 import numpy as np
@@ -11,9 +12,8 @@ from utils import add_arguments, print_arguments, get_feeder_data
 from paddle.fluid.layers.learning_rate_scheduler import _decay_step_counter
 from paddle.fluid.initializer import init_on_cpu
 
-SEED = 90
-# random seed must set before configuring the network.
-fluid.default_startup_program().random_seed = SEED
+if 'ce_mode' in os.environ:
+    np.random.seed(10)
 
 parser = argparse.ArgumentParser(description=__doc__)
 add_arg = functools.partial(add_arguments, argparser=parser)
@@ -87,6 +87,10 @@ def train(args):
     if args.use_gpu:
         place = fluid.CUDAPlace(0)
     exe = fluid.Executor(place)
+
+    if 'ce_mode' in os.environ:
+        fluid.default_startup_program().random_seed = 90
+
     exe.run(fluid.default_startup_program())
 
     if args.init_model is not None:
-- 
GitLab