StarMirror/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2024-03-22 13:10:55 +08:00
Code Issues Projects Releases Wiki Activity

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
Show Stats 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
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):
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
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):
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
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):
def on_build(self, in_ch, ch, n_downscales, kernel_size, dilations=1, subpixel=True):