863a9dcee691e850e79d97a16abb977e24324fa1,tests/convolution_test.py,ConvTest,test_convlayer_3d_bn_reg_dropout_valid_shape,#ConvTest#,146

Before Change

            self.assertAllClose((2, 8, 8, 8, 10), out_3d.shape)

    def test_convlayer_3d_bn_reg_dropout_valid_shape(self):
        x_3d = self.get_3d_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_3d, is_training=True, keep_prob=0.4)
        print(conv_reg)
        with self.test_session() as sess:
            sess.run(tf.global_variables_initializer())
            out_3d = sess.run(conv_reg_out)
            self.assertAllClose((2, 14, 14, 14, 10), out_3d.shape)

    // 2d tests
    def test_2d_conv_default_shape(self):
        x_2d = self.get_2d_input()
        conv_2d = ConvLayer(10, 3, 1)
        conv_2d_out = conv_2d(x_2d)

After Change

                                           dropout_prob=0.4)

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

Instances