github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-08 05:51:40 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

adding kernel initializer option to FUNITAdain block

Browse source
This commit is contained in:
Colombo 2019-09-24 21:30:28 +04:00
parent deeb98474b
commit 4a2203cc35
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
nnlib/FUNIT.py
View file

@ -258,10 +258,10 @@ class FUNIT(object):
inp, mlp = input
x = inp
x = Conv2D(dim, 3, strides=1, padding='valid')(ZeroPadding2D(1)(x))
x = FUNITAdain()([x,mlp])
x = FUNITAdain(kernel_initializer='he_normal')([x,mlp])
x = ReLU()(x)
x = Conv2D(dim, 3, strides=1, padding='valid')(ZeroPadding2D(1)(x))
x = FUNITAdain()([x,mlp])
x = FUNITAdain(kernel_initializer='he_normal')([x,mlp])
return Add()([x,inp])
return func