Skip to content

Commit ed748da

Browse files
authored
"fix dtype test in ci" (#10667)
* "fix ci" * "fix network config" * "rerun ci"
1 parent 48ac978 commit ed748da

File tree

1 file changed

+15
-20
lines changed

1 file changed

+15
-20
lines changed

python/paddle/fluid/tests/unittests/test_network_with_dtype.py

Lines changed: 15 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -24,48 +24,43 @@
2424

2525

2626
class TestNetWithDtype(unittest.TestCase):
27-
def set_network(self):
27+
def setUp(self):
2828
self.dtype = "float64"
2929
self.init_dtype()
30-
main = fluid.Program()
31-
with fluid.program_guard(main):
32-
self.x = fluid.layers.data(name='x', shape=[13], dtype=self.dtype)
33-
self.y = fluid.layers.data(name='y', shape=[1], dtype=self.dtype)
34-
y_predict = fluid.layers.fc(input=self.x, size=1, act=None)
3530

36-
cost = fluid.layers.square_error_cost(input=y_predict, label=self.y)
31+
def run_net_on_place(self, place):
32+
main = fluid.Program()
33+
startup = fluid.Program()
34+
with fluid.program_guard(main, startup):
35+
x = fluid.layers.data(name='x', shape=[13], dtype=self.dtype)
36+
y = fluid.layers.data(name='y', shape=[1], dtype=self.dtype)
37+
y_predict = fluid.layers.fc(input=x, size=1, act=None)
38+
cost = fluid.layers.square_error_cost(input=y_predict, label=y)
3739
avg_cost = fluid.layers.mean(cost)
38-
self.program = main
39-
self.fetch_list = [avg_cost]
40+
sgd_optimizer = fluid.optimizer.SGD(learning_rate=0.001)
41+
sgd_optimizer.minimize(avg_cost)
4042

41-
sgd_optimizer = fluid.optimizer.SGD(learning_rate=0.001)
42-
sgd_optimizer.minimize(avg_cost)
43-
44-
def run_net_on_place(self, place):
43+
fetch_list = [avg_cost]
4544
train_reader = paddle.batch(
4645
paddle.dataset.uci_housing.train(), batch_size=BATCH_SIZE)
47-
feeder = fluid.DataFeeder(place=place, feed_list=[self.x, self.y])
46+
feeder = fluid.DataFeeder(place=place, feed_list=[x, y])
4847
exe = fluid.Executor(place)
49-
exe.run(fluid.default_startup_program())
48+
exe.run(startup)
5049
for data in train_reader():
51-
exe.run(self.program,
52-
feed=feeder.feed(data),
53-
fetch_list=self.fetch_list)
50+
exe.run(main, feed=feeder.feed(data), fetch_list=fetch_list)
5451
# the main program is runable, the datatype is fully supported
5552
break
5653

5754
def init_dtype(self):
5855
pass
5956

6057
def test_cpu(self):
61-
self.set_network()
6258
place = fluid.CPUPlace()
6359
self.run_net_on_place(place)
6460

6561
def test_gpu(self):
6662
if not core.is_compiled_with_cuda():
6763
return
68-
self.set_network()
6964
place = fluid.CUDAPlace(0)
7065
self.run_net_on_place(place)
7166

0 commit comments

Comments
 (0)