diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-06-08 21:52:43 +0300 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-06-08 21:52:43 +0300 |
| commit | 16e5e41530ff5e399b13fd7d36a0a2bffff0454a (patch) | |
| tree | 374ff3cf44b9c69e5b86612c9e77921dcfd717b5 /test | |
| parent | 127c56b1ec59a36d7b204cad799d070069ecdde9 (diff) | |
Natalia fix corner case where both reference impl and cudnn produce nans
Diffstat (limited to 'test')
| -rw-r--r-- | test/test.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test.lua b/test/test.lua index 3d2521e..a448317 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1291,7 +1291,7 @@ end function cudnntest.BatchNormalization() local size = { - math.random(1, 32), + math.random(2, 32), math.random(16, 256), } testBatchNormalization('BatchNormalization', size) |