Update modeling_custom.py

modeling_custom.py

@@ -48,7 +48,7 @@ class GatingNetwork(nn.Module):
         # self.dropout_list = [nn.Dropout(dropout_rate), nn.Dropout(dropout_rate)]
         layers.append(nn.Linear(in_features, out_features, bias=bias))
         self.layers = nn.ModuleList(layers)
-        print("Gating network layers:", self.layers)
+        # print("Gating network layers:", self.layers)
 
     def forward(self, x: torch.FloatTensor) -> torch.FloatTensor:
         orig_shape = x.shape