commit | 5221745c2122f7b5fc536bc5076702f214a3a96a | [log] [tgz] |
---|---|---|
author | Eli Stevens <elis@doselab.com> | Wed Feb 15 04:26:44 2017 -0800 |
committer | soumith <soumith@fb.com> | Wed Feb 15 04:26:44 2017 -0800 |
tree | 943441ec67eeb52a77491d3e7ea18100d2d71480 | |
parent | 000ca44b1615669b0643c07ce4750e902c7a1af8 [diff] |
add test for bias=False for 3d convolution
diff --git a/test/test_nn.py b/test/test_nn.py index 89c8d34..3a71bf5 100644 --- a/test/test_nn.py +++ b/test/test_nn.py
@@ -1633,6 +1633,13 @@ ), dict( module_name='Conv3d', + constructor_args=(3, 4, (2, 3, 4), 1, 0, 1, 1, False), + input_size=(2, 3, 3, 4, 5), + cudnn=True, + desc='no_bias' + ), + dict( + module_name='Conv3d', constructor_args=(3, 4, 2, 2), input_size=(2, 3, 5, 5, 5), cudnn=True,