diff options
| author | SeanNaren <taz838@hotmail.co.uk> | 2016-06-11 14:55:30 +0300 |
|---|---|---|
| committer | SeanNaren <taz838@hotmail.co.uk> | 2016-06-11 14:55:30 +0300 |
| commit | de14f7378cc53b5a399a7ca2bc61b9283d82991c (patch) | |
| tree | 45648aafc3e6d6ec585338c716169134118b333a /test | |
| parent | ae1c144739c18b50169717d8517b3b180dbc3c93 (diff) | |
Modified tests and params, updated docs for clipped ReLU
Diffstat (limited to 'test')
| -rw-r--r-- | test/test.lua | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/test/test.lua b/test/test.lua index aac45b4..47305ae 100644 --- a/test/test.lua +++ b/test/test.lua @@ -965,10 +965,11 @@ end function cudnntest.ClippedReLU_single() local input = torch.randn(1, 32):cuda() local ceiling = 0.1 - local module = cudnn.ClippedReLU(true, ceiling):cuda() + local module = cudnn.ClippedReLU(ceiling):cuda() local output = module:forward(input) local expectedOutput = input:clone() expectedOutput[expectedOutput:ge(ceiling)] = ceiling + expectedOutput[expectedOutput:le(0)] = 0 mytester:assertTensorEq(output, expectedOutput) end |