CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages 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
Download Patch File Download Diff File Expand all files Collapse all files
core/leras/archis/DeepFakeArchi.py
+3 -3
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):