search

xxradon / PytorchToCaffe

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs
MIT License
783 stars 224 forks source link

GPU memory are large #25

Closed Wallace00V closed 5 years ago

Wallace00V commented 5 years ago

when i translate mobilenet/mobilenetv2 to caffe，and load the caffe model in python, the GPU memory is huge; even the mobilenetv2 with width multi of 0.25

Wallace00V commented 5 years ago

Answer found: add engine: Caffe in convolution layer