|
|
|
@ -172,12 +172,18 @@ class ParallelOpTest(BaseParallelForTest):
|
|
|
|
|
loss = fluid.layers.mean(x=hidden)
|
|
|
|
|
yield loss
|
|
|
|
|
|
|
|
|
|
def test_fc_with_tiny_data(self):
|
|
|
|
|
def test_simple_fc(self):
|
|
|
|
|
self.run_test(
|
|
|
|
|
callback=self.__network__,
|
|
|
|
|
feed={'img': numpy.random.random(size=(8, 784)).astype('float32')},
|
|
|
|
|
fetch=['fc1.w@GRAD'])
|
|
|
|
|
|
|
|
|
|
def test_fc_with_tiny_data(self):
|
|
|
|
|
self.run_test(
|
|
|
|
|
callback=self.__network__,
|
|
|
|
|
feed={'img': numpy.random.random(size=(1, 784)).astype('float32')},
|
|
|
|
|
fetch=['fc1.w@GRAD'])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class ParallelOpTestMultipleInput(BaseParallelForTest):
|
|
|
|
|
@staticmethod
|
|
|
|
|