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
......@@ -340,5 +340,5 @@ def test_fps_with_dist():
expected_idx = torch.from_numpy(fps_idx).cuda()
features_for_fps_distance = torch.from_numpy(
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)
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