in augmentation/augmenters.py [0:0]
def __init__(self, from_model_name='transformer.wmt19.en-de', to_model_name='transformer.wmt19.de-en',
from_model_checkpt='model1.pt', to_model_checkpt='model1.pt', tokenizer='moses', bpe='fastbpe',
is_load_from_github=True, name='BackTranslationAug', device='cpu', force_reload=False, verbose=0,
n_predictions :int = 1, generation_kwargs = None):
super().__init__(from_model_name='transformer.wmt19.en-de', to_model_name='transformer.wmt19.de-en',
from_model_checkpt='model1.pt', to_model_checkpt='model1.pt', tokenizer='moses', bpe='fastbpe',
is_load_from_github=True, name='BackTranslationAug', device='cpu', force_reload=False, verbose=0)
self.n_predictions = n_predictions
self.generation_kwargs = generation_kwargs or {}