首页 > 解决方案 > 在内存中加载 caffe 模型

问题描述

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

标签: deep-learningcaffe

解决方案


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


推荐阅读