Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
a3c0e67c
Unverified
Commit
a3c0e67c
authored
Jan 28, 2021
by
Nicolas Hug
Committed by
GitHub
Jan 28, 2021
Browse files
increase tol (#3320)
Co-authored-by:
Vasilis Vryniotis
<
datumbox@users.noreply.github.com
>
parent
9aa07224
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
test/test_ops.py
test/test_ops.py
+1
-2
No files found.
test/test_ops.py
View file @
a3c0e67c
...
...
@@ -549,7 +549,7 @@ class DeformConvTester(OpTester, unittest.TestCase):
out_channels
=
2
kernel_size
=
(
3
,
2
)
groups
=
2
tol
=
1
e-3
if
dtype
is
torch
.
half
else
1e-5
tol
=
2
e-3
if
dtype
is
torch
.
half
else
1e-5
layer
=
ops
.
DeformConv2d
(
in_channels
,
out_channels
,
kernel_size
,
stride
=
stride
,
padding
=
padding
,
dilation
=
dilation
,
groups
=
groups
).
to
(
device
=
x
.
device
,
dtype
=
dtype
)
...
...
@@ -651,7 +651,6 @@ class DeformConvTester(OpTester, unittest.TestCase):
@
unittest
.
skipIf
(
not
torch
.
cuda
.
is_available
(),
"CUDA unavailable"
)
def
test_autocast
(
self
):
set_rng_seed
(
0
)
for
dtype
in
(
torch
.
float
,
torch
.
half
):
with
torch
.
cuda
.
amp
.
autocast
():
self
.
_test_forward
(
torch
.
device
(
"cuda"
),
False
,
dtype
=
dtype
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment