机器学习笔记:深度学习模型复杂度

1 时间复杂度(模型运算次数)

  • 决定了模型的训练/预测时间
  • 用FLOPs指代
    • 机器学习笔记:FLOPs_flop 机器学习_UQI-LIUWJ的博客-CSDN博客

2 空间复杂度(模型参数数量)

决定了模型的参数数量

2.1 全连接层参数量

机器学习笔记:深度学习模型复杂度_第1张图片

包含bias的话,参数量就是  (2D+1)*4D+(4D+1)*D

2.2 卷积参数量

(K_H \cdot K_W \cdot C_{in}+1)\cdot C_{out}

 

机器学习笔记:深度学习模型复杂度_第2张图片

你可能感兴趣的:(机器学习,机器学习,深度学习,笔记)