Update modeling_gain_dann.py

modeling_gain_dann.py

@@ -236,7 +236,7 @@ class Network:
 #----------------------------------------------------------------------------------------------    
 
 class GAIN_DANN(nn.Module):
-    def __init__(self, input_dim: int, latent_dim: int, n_class: int, params: Params, metrics: Metrics):
+    def __init__(self, input_dim: int, latent_dim: int, n_class: int, params: Params, metrics: Metrics, hint_rate = 0.9):
         super(GAIN_DANN, self).__init__()
         self.encoder = Encoder(input_dim=input_dim, hidden_dim=128, latent_dim=latent_dim)