training-mfu-calculator
Installation
SKILL.md
MFU计算工具
何时使用本 Skill
- 用户需要计算MFU:评估大模型训练的硬件利用率。
- 用户提到FLOPs、吞吐量、训练效率:分析模型训练性能。
- 用户需要性能评估:对比不同硬件或配置的训练效率。
- 用户提到硬件利用率、算力利用率:优化训练系统性能。
- 用户需要模型性能分析:分析Dense或MoE模型的计算量。
快速开始
最简单的使用方式
from mfu_calculator import MODEL_CONFIGS, MFUCalculator, TrainingConfig