fix dimensions error

This commit is contained in:
iperov 2020-12-11 14:34:37 +04:00
commit 35945b257c
2 changed files with 2 additions and 2 deletions

View file

@ -87,7 +87,7 @@ class DeepFakeArchi(nn.ArchiBase):
return res // (2**4)
def get_out_ch(self):
return 512
return self.e_ch * 8
lowest_dense_res = resolution // (32 if 'd' in opts else 16)