Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove tab

Browse Source
This commit is contained in:
Yin Li 2021-03-26 12:50:49 -04:00
parent abb16fe26a
commit 4898724ac1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
map2map/models/style.py
View File

@ -158,7 +158,7 @@ class ConvStyled3d(nn.Module):
x = x.reshape(1, N * Cin, *DHWin)
x = self.conv(x, w, bias=self.bias, stride=self.stride, groups=N)
_, _, *DHWout = x.shape
x = x.reshape(N, Cout, *DHWout)
x = x.reshape(N, Cout, *DHWout)
return x