Merge pull request #1157 from wanghaoshuang/ce_icnet

fix ce of icnet

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

fluid/icnet/_ce.py

@@ -7,7 +7,7 @@ from kpi import CostKpi, DurationKpi, AccKpi
 
 # NOTE kpi.py should shared in models in some way!!!!
 
-train_cost_kpi = CostKpi('train_cost', 0.02, actived=True)
+train_cost_kpi = CostKpi('train_cost', 0.05, actived=True)
 train_duration_kpi = DurationKpi('train_duration', 0.06, actived=True)
 
 tracking_kpis = [

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: