本文分享模型训练及推理时性能、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版)