[Bug] Correcting directory for model card's translation examples

In the examples generated from clicking the Train icon for translation models,

It’s pointing to a deprecated directory when using the v4.17 example script path, i.e. ./examples/pytorch/seq2seq:

# git configuration to download our fine-tuning script
git_config = {'repo': 'https://github.com/huggingface/transformers.git','branch': 'v4.17.0'}

...

# creates Hugging Face estimator
huggingface_estimator = HuggingFace(
	entry_point='run_translation.py',
	source_dir='./examples/pytorch/seq2seq',
	instance_type='ml.p3.2xlarge',
	instance_count=1,
	role=role,
	git_config=git_config,
	transformers_version='4.17.0',
	pytorch_version='1.10.2',
	py_version='py38',
	hyperparameters = hyperparameters
)

From the github site, it should be pointing to https://github.com/huggingface/transformers/tree/v4.17.0/examples/pytorch/translation, the generated example from the Train button should be appropriately changed to

source_dir='./examples/pytorch/translation'

thanks for reporting! for @philschmid i think :slight_smile:

1 Like