提交 58b98660 编写于 作者: Y Yibing Liu

Use double quotes for environ variable

上级 3f3b5784
......@@ -55,7 +55,7 @@ def train(train_reader,
feeder = fluid.DataFeeder(feed_list=[data, label], place=place)
# For internal continuous evaluation
if 'CE_MODE_X' in os.environ:
if "CE_MODE_X" in os.environ:
fluid.default_startup_program().random_seed = 110
exe.run(fluid.default_startup_program())
for pass_id in xrange(pass_num):
......@@ -80,7 +80,7 @@ def train(train_reader,
pass_end = time.time()
# For internal continuous evaluation
if 'CE_MODE_X' in os.environ:
if "CE_MODE_X" in os.environ:
print("kpis train_acc %f" % avg_acc)
print("kpis train_cost %f" % avg_cost)
print("kpis train_duration %f" % (pass_end - pass_start))
......
......@@ -65,7 +65,7 @@ def prepare_data(data_type="imdb",
raise RuntimeError("No such dataset")
if data_type == "imdb":
if 'CE_MODE_X' in os.environ:
if "CE_MODE_X" in os.environ:
train_reader = paddle.batch(
paddle.dataset.imdb.train(word_dict), batch_size=batch_size)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册