We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent db195af commit 85468daCopy full SHA for 85468da
tests/links_tests/update_tests/test_film_update.py renamed to tests/links_tests/update_tests/test_gnn_film_update.py
@@ -65,17 +65,17 @@ def test_backward_cpu(update, data):
65
# type: (GNNFiLMUpdate, Tuple[numpy.ndarray, numpy.ndarray, numpy.ndarray]) -> None # NOQA
66
atom_data, adj_data, y_grad = data
67
gradient_check.check_backward(
68
- update, (atom_data, adj_data), y_grad, atol=1e-3, rtol=1e-3)
+ update, (atom_data, adj_data), y_grad, atol=1e-2, rtol=1e-2)
69
70
71
@pytest.mark.gpu
72
def test_backward_gpu(update, data):
73
74
atom_data, adj_data, y_grad = map(cuda.to_gpu, data[:3])
75
update.to_gpu()
76
- print(type(adj_data))
+ # print(type(adj_data))
77
78
79
80
81
def test_forward_cpu_graph_invariant(update, data):
0 commit comments