Skip to content

Commit 5b06944

Browse files
Yanceydzhwinter
authored andcommitted
fix trainer import error on ce (#10448)
* fix trainer import error on ce * fix setup.py.in
1 parent dce0732 commit 5b06944

File tree

3 files changed

+4
-1
lines changed

3 files changed

+4
-1
lines changed

python/paddle/fluid/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@
6060
'io',
6161
'initializer',
6262
'layers',
63+
'transpiler'
6364
'nets',
6465
'optimizer',
6566
'learning_rate_decay',

python/paddle/fluid/trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
import data_feeder
2020
import contextlib
2121
import io
22+
import transpiler
2223

2324
# optimizer is same as the parameter of Trainer.__init__. Rename it to opt_module
2425
import optimizer as opt_module

python/setup.py.in

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,8 @@ packages=['paddle',
6868
'paddle.fluid',
6969
'paddle.fluid.proto',
7070
'paddle.fluid.proto.profiler',
71-
'paddle.fluid.layers']
71+
'paddle.fluid.layers',
72+
'paddle.fluid.transpiler']
7273

7374
if '${WITH_FLUID_ONLY}'== 'OFF':
7475
packages+=['paddle.proto',

0 commit comments

Comments
 (0)