Unverified Commit 040658b7 authored by ChaimZhu's avatar ChaimZhu Committed by GitHub
Browse files

[Fix] Fix features variable may be None in points sampler (#2234)

parent cdfbdc0b
......@@ -102,7 +102,7 @@ class PointsSampler(nn.Module):
"""
if points_xyz.dtype == torch.half:
points_xyz = points_xyz.to(torch.float32)
if features.dtype == torch.half:
if features is not None and features.dtype == torch.half:
features = features.to(torch.float32)
indices = []
......
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