pytorch_translate/dual_learning/dual_learning_models.py
pytorch_translate/word_prediction/word_prediction_model.py