Update modeling_autoencoder.py

modeling_autoencoder.py

@@ -202,7 +202,7 @@ def create_layers(
             layers.append(nn.Linear(input_dim, output_dim))
         elif layer_types in ['lstm', 'rnn', 'gru']:
             rnn_layer = getattr(nn, layer_types.upper())  # Dynamically get the RNN layer class.
-            half_output_dim = output_dim // (2 if bidirectional else 1)
+            half_output_dim = output_dim // (1 if bidirectional else 1)
             layers.append(rnn_layer(input_dim, half_output_dim, batch_first=True, bidirectional=bidirectional))
 
         # Add dropout layer between layers, except for the last layer.