Update modeling_autoencoder.py

modeling_autoencoder.py

@@ -189,7 +189,10 @@ def create_layers(
 
         # Set the final layer's dimension to classes if specified and valid.
         if isinstance(classes, int) and not isinstance(classes, bool):
-            output_dimensions[-1] = classes
+            if bidirectional:
+                output_dimensions[-1] = classes//2
+            else: 
+                output_dimensions[-1] = classes
 
         # Adjust dimensions for bidirectional RNN layers.
         if bidirectional and (layer_types in ['lstm', 'rnn', 'gru']):