fix dimensions error

This commit is contained in:
iperov committed 2020-12-11 14:34:37 +04:00
1 parent ca03bbca04
commit 35945b257c
2 files changed
+2 -2

No files matched your search

+1 -1
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)