train_on_texts gets stuck before starting epochs 