在内存中加载caffe模型

时间:2019-07-04 10:29:08

标签: deep-learning caffe

每次训练时都必须加载模型。模型的加载大约需要3.9秒。我想一次将模型加载到内存中,以便可以连续使用它进行训练。

1 个答案:

答案 0 :(得分:0)

caffe/build/tools/caffe train --solver solver_file.prototxt --weights pretrained_model.caffemodel -gpu 0

在上面的代码中,您应该使用模型名称而不是pretrained_model,然后,在训练开始时,将加载该模型,并且不会再次加载该模型。

https://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.html