Pytorch深度学习模型的指标测试【1】(如:模型大小、模型操作量)

引言: 几种轻量级深度学习模型的保存文件大小和模型的操作量统计。


Pytorch深度学习模型的指标测试【1】(如:模型大小、模型操作量)

  • 1. 测试模型说明
  • 2 模型结构
    • 2.1 普通单层卷积模型
    • 2.2 普通单层 Separable Conv 模型
    • 2.3 Siamese 模型

1. 测试模型说明

  • 单层卷积模型
  • 单层可分离卷积模型
  • Siamese 孪生模型

测试代码:

#########################################################################################################################
model1 = net1()
model2 = net2

你可能感兴趣的:(风机关键部件以及轴承故障诊断,计算机视觉,-,Opencv,强化学习等的,趣味小实验,深度学习,pytorch,人工智能)