Important

You are viewing the NeMo 2.0 documentation. This release introduces significant changes to the API and a new library, NeMo Run. We are currently porting all features from NeMo 1.0 to 2.0. For documentation on previous versions or features not yet available in 2.0, please refer to the NeMo 24.07 documentation.

Performance#

Training Performance Results#

We measured the training throughput of the NSFW model on a single DGX A100 node. The table below shows the performance results.

  • NVIDIA DGX A100 (8 x A100 80GB for CLIP L/14 base model)

Model

Metric

Nodes

1

NSFW L/14

Samples per Second

1680

Inference Performance Results#

Because the NSFW model is built on top of pretrained CLIP with a small linear probing layer, inference performance is expected to match CLIP inference performance. See CLIP Inference Performance Results for details.