pre train steps - 3400 vocab size - 5k