

如何计算模型大小

定义

通过计算模型参数数目，有助于衡量模型所需内存。利用参数数目计算模型大小

$$Size = \frac {Num \times 4.0}{1024\times 1024}$$

• 模型大小的单位是MB
• 通常使用32位浮点数作为数据类型

实现

• 实现文件：metrics/model.py
• 测试文件：test_params_flops.py