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

Optional 2

Browse source
This commit is contained in:
TalosOfCrete 2020-06-08 12:53:56 -05:00
commit 13b53a6e14
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
core/leras/archis/DeepFakeArchi.py
View file

@ -159,7 +159,7 @@ class DeepFakeArchi(nn.ArchiBase):
return self.ae_out_ch
class Decoder(nn.ModelBase):
def on_build(self, in_ch, d_ch, d_mask_ch, is_hd ):
def on_build(self, in_ch, d_ch, d_mask_ch, is_hd, separable=False ):
self.is_hd = is_hd
self.upscale0 = Upscale(in_ch, d_ch*8, kernel_size=3, separable=separable)