How to save and load models in Caffe?

In Caffe, you can use the following commands to save and load models:

Save model:

./build/tools/caffe train -solver /path/to/solver.prototxt -snapshot /path/to/snapshot_iter_1000.solverstate

Load model:

./build/tools/caffe train -solver /path/to/solver.prototxt -weights /path/to/pretrained_model.caffemodel

Among them, /path/to/solver.prototxt is the configuration file, /path/to/snapshot_iter_1000.solverstate is the saved model state file, and /path/to/pretrained_model.caffemodel is the pre-trained model file. These commands make it easy to save and load models.

Leave a Reply 0

Your email address will not be published. Required fields are marked *


广告
Closing in 10 seconds
bannerAds