facebookarchive / caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.
https://caffe2.ai
Apache License 2.0
8.42k stars 1.94k forks source link

caffemodel translate success in MNIST and cifar but failed in CaffeNet #1844

Open DevilGragon opened 6 years ago

DevilGragon commented 6 years ago

Hi all, I want to translate caffemodel to caffe2 model following the guide using the command line:

  python -m caffe2.python.caffe_translator deploy.prototxt pretrained.caffemodel

and get right results in MNIST and CIFAR-10 but failed in CaffeNet. The progress was killed by Ubuntu automatically and corrupted without any result.

How do i make it right? Is my PC too old?

Thanks!

dongwflj commented 6 years ago

I get the same problem, can anyone help? python -m caffe2.python.caffe_translator deploy2.prototxt EmotiW_VGG_S.caffemodel WARNING:root:This caffe2 python run does not have GPU support. Will run in CPU only mode. WARNING:root:Debug message: No module named caffe2_pybind11_state_gpu INFO:caffe_translator:Translate layer input INFO:caffe_translator:Translate layer conv1 INFO:caffe_translator:Translate layer relu1 INFO:caffe_translator:Translate layer norm1 INFO:caffe_translator:Translate layer pool1 INFO:caffe_translator:Translate layer conv2 INFO:caffe_translator:Translate layer relu2 INFO:caffe_translator:Translate layer pool2 INFO:caffe_translator:Translate layer conv3 INFO:caffe_translator:Translate layer relu3 INFO:caffe_translator:Translate layer conv4 INFO:caffe_translator:Translate layer relu4 INFO:caffe_translator:Translate layer conv5 INFO:caffe_translator:Translate layer relu5 INFO:caffe_translator:Translate layer pool5 INFO:caffe_translator:Translate layer fc6 Killed