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

fix

Browse source
This commit is contained in:
Colombo 2020-04-07 22:06:32 +04:00
parent d8a792232c
commit 935d940ace
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -39,7 +39,7 @@ class DeepFakeArchi(nn.ArchiBase):
return x return x
def get_out_ch(self): def get_out_ch(self):
return (self.out_ch // 4) * 4 return (self.out_ch // 4) * 4 if self.subpixel else self.out_ch
class DownscaleBlock(nn.ModelBase): class DownscaleBlock(nn.ModelBase):
def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True): def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True):
@ -227,7 +227,7 @@ class DeepFakeArchi(nn.ArchiBase):
return x return x
def get_out_ch(self): def get_out_ch(self):
return (self.out_ch // 4) * 4 return (self.out_ch // 4) * 4 if self.subpixel else self.out_ch
class DownscaleBlock(nn.ModelBase): class DownscaleBlock(nn.ModelBase):
def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True): def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True):
@ -359,7 +359,7 @@ class DeepFakeArchi(nn.ArchiBase):
return x return x
def get_out_ch(self): def get_out_ch(self):
return (self.out_ch // 4) * 4 return (self.out_ch // 4) * 4 if self.subpixel else self.out_ch
class DownscaleBlock(nn.ModelBase): class DownscaleBlock(nn.ModelBase):
def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True): def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True):