← 返回课程库
Runnable starter
参数量、FLOPs 与显存估算器
用纯 Python 函数估算 dense Transformer 参数量、6ND 训练 FLOPs 与不同位宽的权重显存。
环境
Python 3.11+，无第三方依赖
文件
运行与验收
python -m unittest -v test_resource_estimates.py先确保测试通过，再修改输入、增加失败案例，并把运行输出保留在自己的实验记录中。