test_keras2_numeric.py 文件源码

def test_large_batch_gpu(self):

        batch_size = 2049
        num_channels = 4
        kernel_size = 3

        model = Sequential()
        model.add(TimeDistributed(Dense(num_channels), input_shape=(batch_size, kernel_size)))

        model.set_weights([(np.random.rand(*w.shape)-0.5)*0.2 for w in model.get_weights()])

        self._test_keras_model(model, input_blob='data', output_blob='output', delta=1e-2)