diff --git a/core/leras/archis/DeepFakeArchi.py b/core/leras/archis/DeepFakeArchi.py
index e2ee408..9818ab8 100644
--- a/core/leras/archis/DeepFakeArchi.py
+++ b/core/leras/archis/DeepFakeArchi.py
@@ -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)
 
diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py
index 4a227ed..e8b9a06 100644
--- a/models/Model_SAEHD/Model.py
+++ b/models/Model_SAEHD/Model.py
@@ -248,7 +248,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ...
 
                 if self.is_training:
                     if self.options['true_face_power'] != 0:
-                        self.code_discriminator = nn.CodeDiscriminator(ae_dims, code_res=model_archi.Inter.get_out_res(), name='dis' )
+                        self.code_discriminator = nn.CodeDiscriminator(ae_dims, code_res=self.inter.get_out_res(), name='dis' )
                         self.model_filename_list += [ [self.code_discriminator, 'code_discriminator.npy'] ]
 
             elif 'liae' in archi_type: