Skip to content

Commit 1f9aaf8

Browse files
authored
Merge pull request #59 from joaopcm1996/patch-3
Update model parallelism sagemaker-notebook.ipynb
2 parents 6bdcc68 + 2883f1e commit 1f9aaf8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sagemaker/04_distributed_training_model_parallelism/sagemaker-notebook.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -195,7 +195,7 @@
195195
"source": [
196196
"## Creating an Estimator and start a training job\n",
197197
"\n",
198-
"In this example we are going to use the `run_glue.py` from the transformers example scripts. We modified it and included `SageMakerTrainer` instead of the `Trainer` to enable model-parallelism. You can find the code [here](https://github.com/huggingface/transformers/tree/master/examples/text-classification).\n",
198+
"In this example we are going to use the `run_glue.py` from the transformers example scripts. We modified it and included `SageMakerTrainer` instead of the `Trainer` to enable model-parallelism. You can find the code [here](https://github.com/huggingface/transformers/tree/master/examples/pytorch/text-classification).\n",
199199
"\n",
200200
"```python\n",
201201
"from transformers.sagemaker import SageMakerTrainingArguments as TrainingArguments, SageMakerTrainer as Trainer\n",
@@ -483,4 +483,4 @@
483483
},
484484
"nbformat": 4,
485485
"nbformat_minor": 4
486-
}
486+
}

0 commit comments

Comments
 (0)