included Bart's training extension for computing word embeddings