NVIDIA / NVIDIA/TensorRT-LLM

Low MMLU score (86.63) on DeepSeek-R1-FP4 with trtllm-eval on B200, official score ~90.7

Open
#5,154 3 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

triaged
Dominant language
Python
Stars
14.7k
Forks
2.8k
Avg merge
2d 23h
Merged PRs (30d)
489

Description

Hi,

We are running the DeepSeek-R1-FP4 model on a B200 GPU using the official trtllm-eval tool to evaluate MMLU performance. However, the accuracy I get is significantly lower than the official reported FP4 score.

Details:

cat >./extra-llm-api-config.yml <<EOF
use_cuda_graph: true
enable_attention_dp: true
EOF

trtllm-eval --model /data/models/nvidia/DeepSeek-R1-FP4 \
  --tp_size 8 \
  --kv_cache_free_gpu_memory_fraction 0.8 \
  --extra_llm_api_options ./extra-llm-api-config.yml \
  mmlu \
  --dataset_path /data/models/dataset/mmlu/mmlu-un-huggingface/data

What I have verified:

Questions:

  • Is there an official recommended full startup script or configuration for running MMLU on DeepSeek-R1-FP4?
  • Could the accuracy gap be due to missing or incorrect parameters such as number of shots, temperature, or evaluation subsets?
  • Are there known compatibility issues between TensorRT-LLM versions and DeepSeek-R1-FP4 that might affect accuracy?
  • Any suggestions on debugging or tuning to achieve official-level accuracy?

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the DeepSeek V3 evaluation section in examples/models/core/deepseek_v3/README.md and the supplied trtllm-eval command and extra-llm-api-config.yml. Reproduce the three reported MMLU runs using the stated model, dataset, release:0.21.0rc0 image, and TP8 configuration, then compare evaluation parameters with the official FP4 setup. Done means identifying the cause of the accuracy gap or documenting a confirmed compatibility or configuration issue.

Written by the indexing model from the issue text.

Assessment

Tech stack
docker, python, pytorch
Domain
machine-learning, performance, testing-qa
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.