AI实战：pytorch、 tensorflow 对比之推理时性能、GPU占用对比（一）：DenseNet

本文分享模型训练及推理时性能、GPU占用对比：pytorch 与 tensorflow 上的对比。

DenseNet定义

论文链接：Densely Connected Convolutional Networks

参考：深度/机器学习基础知识要点：CNN、ResNet、DenseNet

DenseNet参数

• 参数情况
  

• flops

  

DensetNet201 性能对比

• 图片输入尺寸
  224 × 256

• 环境

  • Ubuntu 16.04
  • torch 1.2.0
  • torchvision 0.4.0
  • tensorflow 1.14.0

• 对比

  平台	训练GPU占用（M）	推理GPU占用（M）	推理速度（100张s）	模型文件大小（M）
  tensorflow		5300	3.1	70
  pytorch	470	600	2.3	73.6

参考

• AI实战：pytorch版DenseNet迁移学习
• AI实战：迁移学习之使用ResNet做分类 (tensorflow版)