提交 5af0c60f 编写于 作者: M minqiyang

Add compat deps

上级 b5de0166
...@@ -109,6 +109,7 @@ def train(use_cuda, train_program, parallel, params_dirname): ...@@ -109,6 +109,7 @@ def train(use_cuda, train_program, parallel, params_dirname):
feed_order=['pixel', 'label']) feed_order=['pixel', 'label'])
else: else:
import paddle.fluid.core as core import paddle.fluid.core as core
import paddle.compat as cpt
try: try:
trainer.train( trainer.train(
reader=train_reader, reader=train_reader,
......
...@@ -92,6 +92,7 @@ def train(use_cuda, train_program, params_dirname, parallel): ...@@ -92,6 +92,7 @@ def train(use_cuda, train_program, params_dirname, parallel):
feed_order=['img', 'label']) feed_order=['img', 'label'])
else: else:
import paddle.fluid.core as core import paddle.fluid.core as core
import paddle.compat as cpt
try: try:
trainer.train( trainer.train(
num_epochs=1, num_epochs=1,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册