Caffe2 model file formats

To be able to use Caffe models in Caffe2, we also need to understand the model file formats that Caffe2 can import. Just like Caffe, Caffe2 also uses Protobuf for serialization and deserialization of its model files. Caffe2 imports a trained model from two files:

1. The structure of the neural network stored as a predict_net.pb file or as a predict_net.pbtxt file
2. The weights of the operators of the neural network stored as a init_net.pb file

predict_net file

The predict_net binary file, which is usually named predict_net.pb, holds the list of operators in the neural network, the parameters of each operator, and the connections between the operators. This file is a serialization of the neural network...