怎么利用imagenet训练googlenet

如题所述

训练配置:batchsize=128
caffe自有的imagenet with cuDNN模型快于googlenet with cuDNN
VGG16层不用cuDNN慢于caffe自有的imagenet with cuDNN模型
VGG19层不用cuDNN慢于caffe自有的imagenet with cuDNN模型

一、CAFFE 自带配置,使用cuDNN
Forward速度 : 220ms
Backward速度 :360ms
二、CAFFE 自带配置,不使用cuDNN

Forward速度 : 300ms
Backward速度 :410ms
三、GoogleNet,使用cuDNN

Forward速度 : 614ms
Backward速度 :1377ms

四、GoogleNet,不使用cuDNN

Forward速度 : 1145ms
Backward速度 :2009ms
五、VGG16层,使用cuDNN

Forward速度 : 3101ms
Backward速度 :8002ms
六、VGG19层,使用cuDNN

Forward速度 : 3972ms
Backward速度 :8540ms
温馨提示:答案为网友推荐,仅供参考
相似回答