Skip to content

GitLab

"docs/source/api/vscode:/vscode.git/clone" did not exist on "acb4eb7ecdf126949821b6b2607db5b77fb42acb"
Unverified Commit 6356cbdc authored by encore-zhou's avatar encore-zhou Committed by GitHub
Browse files

[Fix]: add missing test data for pointnet modules (#85) 

* add test data

* update fps test data

* update fps test data
parent 4040a074
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_pointnet_ops.py
View file @ 6356cbdc
...@@ -340,5 +340,5 @@ def test_fps_with_dist(): ...@@ -340,5 +340,5 @@ def test_fps_with_dist():
expected_idx = torch.from_numpy(fps_idx).cuda() expected_idx = torch.from_numpy(fps_idx).cuda()
features_for_fps_distance = torch.from_numpy( features_for_fps_distance = torch.from_numpy(
features_for_fps_distance).cuda() features_for_fps_distance).cuda()
idx = furthest_point_sample_with_dist(features_for_fps_distance, 32) idx = furthest_point_sample_with_dist(features_for_fps_distance, 16)
assert torch.all(idx == expected_idx) assert torch.all(idx == expected_idx)
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