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

还原SAEHD模型参数

Browse source
This commit is contained in:
cszdlt 2019-12-29 00:53:53 +08:00
commit c104ba6239
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/Model_SAEHD/Model.py
View file

@ -213,7 +213,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 +327,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