Model
NVIDIA Deep Learning Examples
BERT PyTorch checkpoint (Dist-4L-288D, SST-2, seqLen128, AMP)BERT Distilled 4L-288D PyTorch checkpoint distilled on GLUE/SST-2 dataset using AMP
Use the NGC CLI to download:
Copied!
BERT Distilled 4L-288D PyTorch checkpoint distilled on GLUE/SST-2 dataset using AMP