NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
BERT PyTorch checkpoint (Dist-6L-768D, SST-2, AMP)
Model
NVIDIA Deep Learning Examples
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

NameSizeUpdatedActions
config.json
848 BFebruary 28, 2025 UTC
pytorch_model.bin
255.48 MBFebruary 28, 2025 UTC
vocab.txt
226.08 KBFebruary 28, 2025 UTC