Skip to content

GitLab

"examples/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "0ead9141543fea77f98f3aaee31bb1e309b53219"
Commit f445726c authored by rusty1s's avatar rusty1s
Browse files

increase neighbors

parent 314d1637
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/test_radius.py
View file @ f445726c
...@@ -64,7 +64,7 @@ def test_radius_graph_large(dtype, device): ...@@ -64,7 +64,7 @@ def test_radius_graph_large(dtype, device):
x = torch.randn(1000, 3) x = torch.randn(1000, 3)
edge_index = radius_graph(x, r=0.5, flow='target_to_source', loop=True, edge_index = radius_graph(x, r=0.5, flow='target_to_source', loop=True,
max_num_neighbors=1000, num_workers=6) max_num_neighbors=2000, num_workers=6)
tree = scipy.spatial.cKDTree(x.numpy()) tree = scipy.spatial.cKDTree(x.numpy())
col = tree.query_ball_point(x.cpu(), r=0.5) col = tree.query_ball_point(x.cpu(), r=0.5)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment