每次训练时都必须加载模型。模型的加载大约需要3.9秒。我想一次将模型加载到内存中,以便可以连续使用它进行训练。
答案 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