CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

RecycleGAN fixes

Browse Source
This commit is contained in:
iperov
2018-12-28 23:13:29 +04:00
parent 1f86d7f1dd
commit 474fff248f
3 changed files with 25 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
nnlib/nnlib.py
+2 -2
View File
@@ -551,8 +551,8 @@ NLayerDiscriminator = nnlib.NLayerDiscriminator
for i in range(n_blocks):
x = ResnetBlock(ngf*4)(x)
x = ReLU()(XNormalization(Conv2DTranspose(ngf*2, 3, 2, 'same')(x)))
x = ReLU()(XNormalization(Conv2DTranspose(ngf , 3, 2, 'same')(x)))
x = ReLU()(XNormalization(PixelShuffler()(Conv2D(ngf*2 *4, 3, 1, 'same')(x))))
x = ReLU()(XNormalization(PixelShuffler()(Conv2D(ngf *4, 3, 1, 'same')(x))))
x = ReflectionPadding2D((3,3))(x)
x = Conv2D(output_nc, 7, 1, 'valid')(x)