Add NCF to pytorch training benchmark doc (#5352)

* add previous version (25.6)

* fix template

* Formatting and wording fixes

* add caveats

* update yaml

* add note to pytorch-training

* fix template

* make model name shorter
This commit is contained in:
Peter Park
2025-09-16 13:29:28 -04:00
committed by GitHub
parent 5c7ccb3c26
commit bab853a0d3
2 changed files with 18 additions and 0 deletions

View File

@@ -159,3 +159,11 @@ model_groups:
url: https://huggingface.co/black-forest-labs/FLUX.1-dev
precision: BF16
training_modes: [pretrain]
- group: NCF
tag: ncf
models:
- model: NCF
mad_tag: pyt_ncf_training
model_repo:
url: https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Recommendation/NCF
precision: FP32