aglib/meliad/transformer/configs/recurrent/states_1024.gin

# Apply after the gate config.

transformer_layer.TransformerLayer:
  recurrent_num_states = 1024