github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

SAEHD:optimized architecture, you have to restart training

Browse source
This commit is contained in:
Colombo 2019-12-28 16:50:33 +04:00
parent 4e67278c66
commit 28549dc153
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
models/Model_SAEHD/Model.py
View file

@ -198,8 +198,6 @@ class SAEHDModel(ModelBase):
if dims % 2 != 0: if dims % 2 != 0:
dims += 1 dims += 1
def func(x): def func(x):
for i in [8,4,2]: for i in [8,4,2]:
@ -213,7 +211,7 @@ class SAEHDModel(ModelBase):
x = Add()([x, x0]) x = Add()([x, x0])
x = LeakyReLU(0.2)(x) x = LeakyReLU(0.2)(x)
return Conv2D(output_nc, kernel_size=5, padding='same', activation='sigmoid')(x) return Conv2D(output_nc, kernel_size=1, padding='same', activation='sigmoid')(x)
return func return func
@ -327,7 +325,7 @@ class SAEHDModel(ModelBase):
x = Add()([x, x0]) x = Add()([x, x0])
x = LeakyReLU(0.2)(x) x = LeakyReLU(0.2)(x)
return Conv2D(output_nc, kernel_size=5, padding='same', activation='sigmoid')(x) return Conv2D(output_nc, kernel_size=1, padding='same', activation='sigmoid')(x)
return func return func