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

Change PatchGAN architecture

Browse Source
This commit is contained in:
Yin Li 2020-02-09 17:32:46 -05:00
parent 1dd2b6d748
commit a2e3a9dc64
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
map2map/models/patchgan.py
View File

@ -10,9 +10,8 @@ class PatchGAN(nn.Module):
self.convs = nn.Sequential( self.convs = nn.Sequential(
ConvBlock(in_chan, 32, seq='CA'), ConvBlock(in_chan, 32, seq='CA'),
ConvBlock(32, 64, seq='CBA'), ConvBlock(32, 64, seq='CBA'),
ConvBlock(64, seq='CBA'), ConvBlock(64, 128, seq='CBA'),
ConvBlock(64, 32, seq='CBA'), ConvBlock(128, out_chan, seq='C'),
nn.Conv3d(32, out_chan, 1),
) )
def forward(self, x): def forward(self, x):