From d8c7ae96a0e2c88b151d1bc51b94a1133d443b6b Mon Sep 17 00:00:00 2001 From: Pawel Swietojanski Date: Mon, 28 Dec 2015 16:15:04 +0000 Subject: [PATCH] fix to bias shape assert --- mlp/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlp/utils.py b/mlp/utils.py index 569990f..34d62e5 100644 --- a/mlp/utils.py +++ b/mlp/utils.py @@ -348,7 +348,7 @@ def test_conv_linear_pgrads(layer, kernel_order='ioxy', kernels_first=True, assert layer_bias_grads.shape == true_bias_grads.shape, ( 'Layer pgrads gives incorrect shaped bias gradients output. ' 'Correct shape is {0} but returned shape is {1}.' - .format(true_kernel_grads.shape, layer_kernel_grads.shape) + .format(true_bias_grads.shape, layer_bias_grads.shape) ) assert numpy.allclose(layer_bias_grads, true_bias_grads), ( 'Layer pgrads does not give correct bias gradients output. '