提交 003dc79b 编写于 作者: Y Yang Yang

add multiple fc layers

上级 1e4b4d7d
...@@ -179,8 +179,10 @@ class ParallelOpTestMultipleInput(BaseParallelForTest): ...@@ -179,8 +179,10 @@ class ParallelOpTestMultipleInput(BaseParallelForTest):
y = fluid.layers.data(shape=[784], dtype='float32', name='img2', stop_gradient=False) y = fluid.layers.data(shape=[784], dtype='float32', name='img2', stop_gradient=False)
yield [x, y] yield [x, y]
x = x + y x = x + y
hidden = fluid.layers.fc(input=x, size=200, param_attr='fc1.w') hidden1 = fluid.layers.fc(input=x, size=200, param_attr='fc1.w')
loss = fluid.layers.mean(x=hidden) hidden2 = fluid.layers.fc(input=hidden1, size=200, param_attr='fc2.w')
hidden3 = fluid.layers.fc(input=hidden2, size=200, param_attr='fc3.w')
loss = fluid.layers.mean(x=hidden3)
yield loss yield loss
def test_simple_fc(self): def test_simple_fc(self):
...@@ -190,7 +192,7 @@ class ParallelOpTestMultipleInput(BaseParallelForTest): ...@@ -190,7 +192,7 @@ class ParallelOpTestMultipleInput(BaseParallelForTest):
'img1': numpy.random.random(size=(51, 784)).astype('float32'), 'img1': numpy.random.random(size=(51, 784)).astype('float32'),
'img2': numpy.random.random(size=(51, 784)).astype('float32') 'img2': numpy.random.random(size=(51, 784)).astype('float32')
}, },
fetch=['fc1.w@GRAD']) fetch=['fc1.w@GRAD', 'fc2.w@GRAD', 'fc3.w@GRAD'])
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册