Загрузка модели кафе в память - PullRequest
0 голосов
/ 04 июля 2019

Мне нужно загружать модель каждый раз, когда я ее тренирую.Загрузка модели занимает около 3,9 секунды.Я хочу загрузить модель один раз в память, чтобы я мог постоянно использовать ее для тренировок.

1 Ответ

0 голосов
/ 06 июля 2019
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

...