Skip to content
This repository has been archived by the owner on Nov 13, 2024. It is now read-only.

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
@iperov
iperov committed Apr 7, 2020
1 parent d8a7922 commit 935d940
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions core/leras/archis/DeepFakeArchi.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def forward(self, x):
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):
Expand Down Expand Up @@ -227,7 +227,7 @@ def forward(self, x):
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):
Expand Down Expand Up @@ -359,7 +359,7 @@ def forward(self, x):
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):
Expand Down

0 comments on commit 935d940

Please sign in to comment.