Skip to content

GitLab

Commit ad798fb3 authored by wuyuefeng's avatar wuyuefeng
Browse files

reformat

parent c8755fda
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmdet3d/models/middle_encoders/sparse_unet.py
View file @ ad798fb3
...@@ -181,8 +181,8 @@ class SparseUNet(nn.Module): ...@@ -181,8 +181,8 @@ class SparseUNet(nn.Module):
""" """
features = x.features features = x.features
n, in_channels = features.shape n, in_channels = features.shape
assert (in_channels % assert (in_channels % out_channels
out_channels == 0) and (in_channels >= out_channels) == 0) and (in_channels >= out_channels)
x.features = features.view(n, out_channels, -1).sum(dim=2) x.features = features.view(n, out_channels, -1).sum(dim=2)
return x return x
......
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