Model
NVIDIA Deep Learning Examples
BERT PyTorch checkpoint (Dist-6L-768D, SST-2, AMP)BERT Distilled 6L-768D PyTorch checkpoint finetuned on SST-2 dataset using AMP
Use the NGC CLI to download:
Copied!
BERT Distilled 6L-768D PyTorch checkpoint finetuned on SST-2 dataset using AMP