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

Quick96: optimized architecture

Browse source
This commit is contained in:
Colombo 2019-12-27 21:06:45 +04:00
parent a347bb4f15
commit 0833a38bb9
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
models/Model_Quick96/Model.py
View file

@ -16,7 +16,7 @@ class Quick96Model(ModelBase):
super().__init__(*args, **kwargs, super().__init__(*args, **kwargs,
ask_enable_autobackup=False, ask_enable_autobackup=False,
ask_write_preview_history=False, ask_write_preview_history=False,
ask_target_iter=False, ask_target_iter=True,
ask_batch_size=False, ask_batch_size=False,
ask_random_flip=False) ask_random_flip=False)
@ -92,8 +92,8 @@ class Quick96Model(ModelBase):
y = self.upscale(d_dims//2)(y) y = self.upscale(d_dims//2)(y)
y = self.upscale(d_dims//4)(y) y = self.upscale(d_dims//4)(y)
return Conv2D(3, kernel_size=5, padding='same', activation='tanh')(x), \ return Conv2D(3, kernel_size=1, padding='same', activation='tanh')(x), \
Conv2D(1, kernel_size=5, padding='same', activation='sigmoid')(y) Conv2D(1, kernel_size=1, padding='same', activation='sigmoid')(y)
return func return func