Skip to content

Commit 5cc03ca

Browse files
authored
[BERT/PyT] Update pretrained checkpoint links (NVIDIA#660)
* Update pretrained checkpoint links * update link
1 parent 0e6cfbd commit 5cc03ca

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

PyTorch/LanguageModeling/BERT/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -245,7 +245,7 @@ To train your model using mixed or TF32 precision with Tensor Cores or using FP3
245245

246246
2. Download the NVIDIA pretrained checkpoint.
247247

248-
If you want to use a pre-trained checkpoint, visit [NGC](https://ngc.nvidia.com/catalog/models/nvidia:bert_large_pyt_amp_ckpt_pretraining_lamb). This downloaded checkpoint is used to fine-tune on SQuAD. Ensure you unzip the downloaded file and place the checkpoint in the `checkpoints/` folder. For a checkpoint already fine-tuned for QA on SQuAD v1.1 visit [NGC](https://ngc.nvidia.com/catalog/models/nvidia:bert_large_pyt_amp_ckpt_squad_qa1_1).
248+
If you want to use a pre-trained checkpoint, visit [NGC](https://ngc.nvidia.com/catalog/models/nvidia:bert_pyt_ckpt_large_pretraining_amp_lamb/files). This downloaded checkpoint is used to fine-tune on SQuAD. Ensure you unzip the downloaded file and place the checkpoint in the `checkpoints/` folder. For a checkpoint already fine-tuned for QA on SQuAD v1.1 visit [NGC](https://ngc.nvidia.com/catalog/models/nvidia:bert_pyt_ckpt_large_qa_squad11_amp/files).
249249

250250
3. Build BERT on top of the NGC container.
251251
`bash scripts/docker/build.sh`

0 commit comments

Comments
 (0)