863a9dcee691e850e79d97a16abb977e24324fa1,tests/convolution_test.py,ConvTest,test_convlayer_2d_bn_reg_valid_shape,#ConvTest#,287

Before Change

            self.assertAllClose((2, 16, 16, 10), out_2d.shape)

    def test_convlayer_2d_bn_reg_valid_shape(self):
        x_2d = self.get_2d_input()
        conv_reg = ConvolutionalLayer(
            10, 3, 1,
            padding="VALID",
            w_regularizer=regularizers.l2_regularizer(0.5),
            with_bias=False,
            with_bn=True,
            acti_func="prelu")
        conv_reg_out = conv_reg(x_2d, is_training=True, keep_prob=0.4)
        print(conv_reg)
        with self.test_session() as sess:
            sess.run(tf.global_variables_initializer())
            out_2d = sess.run(conv_reg_out)
            self.assertAllClose((2, 14, 14, 10), out_2d.shape)


if __name__ == "__main__":
    tf.test.main()

After Change

                                           is_training=True)

    def test_convlayer_2d_bn_reg_valid_shape(self):
        input_param = {"n_output_chns": 10,
                       "kernel_size": [3, 2],
                       "stride": [2, 3],
                       "with_bias": False,
                       "with_bn": True,
                       "acti_func": "prelu",
                       "padding": "VALID",
                       "w_regularizer": regularizers.l2_regularizer(0.5)}
        self._test_conv_layer_output_shape(rank=2,
                                           param_dict=input_param,
                                           output_shape=(2, 7, 5, 10),
                                           is_training=True)


if __name__ == "__main__":
    tf.test.main()

Instances