Commit ad798fb3 authored by wuyuefeng's avatar wuyuefeng
Browse files

reformat

parent c8755fda
......@@ -181,8 +181,8 @@ class SparseUNet(nn.Module):
"""
features = x.features
n, in_channels = features.shape
assert (in_channels %
out_channels == 0) and (in_channels >= out_channels)
assert (in_channels % out_channels
== 0) and (in_channels >= out_channels)
x.features = features.view(n, out_channels, -1).sum(dim=2)
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