未验证 提交 3ba7b9b5 编写于 作者: L littletomatodonkey 提交者: GitHub

fix regularizer timeout (#26952)

上级 05da5a3c
...@@ -106,9 +106,9 @@ def bow_net(data, ...@@ -106,9 +106,9 @@ def bow_net(data,
label, label,
dict_dim, dict_dim,
is_sparse=False, is_sparse=False,
emb_dim=128, emb_dim=8,
hid_dim=128, hid_dim=8,
hid_dim2=96, hid_dim2=6,
class_dim=2): class_dim=2):
""" """
BOW net BOW net
...@@ -132,8 +132,8 @@ class TestRegularizer(unittest.TestCase): ...@@ -132,8 +132,8 @@ class TestRegularizer(unittest.TestCase):
def setUp(self): def setUp(self):
self.word_dict = paddle.dataset.imdb.word_dict() self.word_dict = paddle.dataset.imdb.word_dict()
reader = paddle.batch( reader = paddle.batch(
paddle.dataset.imdb.train(self.word_dict), batch_size=8)() paddle.dataset.imdb.train(self.word_dict), batch_size=1)()
self.train_data = [next(reader) for _ in range(5)] self.train_data = [next(reader) for _ in range(1)]
def get_places(self): def get_places(self):
places = [core.CPUPlace()] places = [core.CPUPlace()]
...@@ -245,14 +245,14 @@ class TestRegularizer(unittest.TestCase): ...@@ -245,14 +245,14 @@ class TestRegularizer(unittest.TestCase):
sgd.minimize(loss) sgd.minimize(loss)
with fluid.dygraph.guard(): with fluid.dygraph.guard():
input = fluid.dygraph.to_variable( input = fluid.dygraph.to_variable(
np.random.randn(3, 5).astype('float32')) np.random.randn(3, 2).astype('float32'))
paddle.manual_seed(1) paddle.manual_seed(1)
paddle.framework.random._manual_program_seed(1) paddle.framework.random._manual_program_seed(1)
linear1 = fluid.dygraph.Linear( linear1 = fluid.dygraph.Linear(
5, 2, param_attr=fc_param_attr, bias_attr=fc_param_attr) 2, 2, param_attr=fc_param_attr, bias_attr=fc_param_attr)
linear2 = fluid.dygraph.Linear( linear2 = fluid.dygraph.Linear(
5, 2, param_attr=fc_param_attr, bias_attr=fc_param_attr) 2, 2, param_attr=fc_param_attr, bias_attr=fc_param_attr)
loss1 = linear1(input) loss1 = linear1(input)
loss1.backward() loss1.backward()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册