Skip to content

Commit

Permalink
Merge pull request chainer#3933 from hvy/fix-tol-im2col
Browse files Browse the repository at this point in the history
Adjust tolerances for im2col tests
  • Loading branch information
kmaehashi committed Nov 21, 2017
2 parents 85ac317 + 09e1549 commit a054cd6
Showing 1 changed file with 4 additions and 4 deletions.
Expand Up @@ -116,14 +116,14 @@ def setUp(self):
self.ggx = numpy.random.uniform(
size=self.in_shape).astype(self.dtype)

self.check_backward_options = {}
self.check_backward_options = {'atol': 5e-4, 'rtol': 5e-3}
if self.dtype is numpy.float16:
self.check_backward_options.update({'atol': 5e-4, 'rtol': 5e-3})
self.check_backward_options.update({'atol': 1e-3, 'rtol': 1e-2})

self.check_double_backward_options = {}
self.check_double_backward_options = {'atol': 5e-4, 'rtol': 5e-3}
if self.dtype is numpy.float16:
self.check_double_backward_options.update(
{'atol': 5e-4, 'rtol': 5e-3})
{'atol': 1e-3, 'rtol': 1e-2})

def check_backward(self, x, ksize, stride, pad, cover_all, dilate, gy):
def f(x):
Expand Down

0 comments on commit a054cd6

Please sign in to comment.