commented train up to what embedding_layer_state means