fix dimensions error

2020-12-11 14:34:37 +04:00
parent ca03bbca04
commit 35945b257c
2 changed files with 2 additions and 2 deletions
@@ -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)