Skip to content

Commit 85468da

Browse files
author
corochann
committed
fix tolerance
1 parent db195af commit 85468da

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/links_tests/update_tests/test_film_update.py renamed to tests/links_tests/update_tests/test_gnn_film_update.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,17 +65,17 @@ def test_backward_cpu(update, data):
6565
# type: (GNNFiLMUpdate, Tuple[numpy.ndarray, numpy.ndarray, numpy.ndarray]) -> None # NOQA
6666
atom_data, adj_data, y_grad = data
6767
gradient_check.check_backward(
68-
update, (atom_data, adj_data), y_grad, atol=1e-3, rtol=1e-3)
68+
update, (atom_data, adj_data), y_grad, atol=1e-2, rtol=1e-2)
6969

7070

7171
@pytest.mark.gpu
7272
def test_backward_gpu(update, data):
7373
# type: (GNNFiLMUpdate, Tuple[numpy.ndarray, numpy.ndarray, numpy.ndarray]) -> None # NOQA
7474
atom_data, adj_data, y_grad = map(cuda.to_gpu, data[:3])
7575
update.to_gpu()
76-
print(type(adj_data))
76+
# print(type(adj_data))
7777
gradient_check.check_backward(
78-
update, (atom_data, adj_data), y_grad, atol=1e-3, rtol=1e-3)
78+
update, (atom_data, adj_data), y_grad, atol=1e-2, rtol=1e-2)
7979

8080

8181
def test_forward_cpu_graph_invariant(update, data):

0 commit comments

Comments
 (0)