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

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
models/Model_SAEHD/Model.py
+2 -4
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