سلام
من میخام در Caffe بدلیل کم بودن حافظه پردازنده گرافیکی از مد CPU استفاده کنم(من برای شبکه های کوچکتر تونستم ازGPUبدون هیچ مشکلی استفاده کنم) . ولی وقتی در فایل solver
solver_mode: CPU
پارامتر بالا رو تغییر میدم بازم برای آموزش و تست از GPUاستفاده میشه. لطفا اگر اطلاع دارین راهنمایی بفرمایید چطوری از مد CPU و رم اصلی سیستم استفاده کنم؟
سپاسگزارم.
Log file created at:
> 2017/02/02 17:43:39 Running on machine: ahp Log line format:
> [IWEF]mmdd hh:mm:ss.uuuuuu threadid file:line] msg F0202
> 17:43:39.173938 25656 syncedmem.cpp:56] Check failed: error ==
> cudaSuccess (2 vs. 0) out of memory
------------------------------------------------------------------------
train_net: "voc12/config/VGG-16/train_train.prototxt"
lr_policy: "poly"
power: 0.9
base_lr: 1e-3
average_loss: 20
display: 20
max_iter: 20000
momentum: 0.9
weight_decay: 0.0005
snapshot: 20000
snapshot_prefix: "voc12/model/VGG-16/train"
solver_mode: CPU
ادیتور کدها رو خوب نشون نمیده!
Running /home/ahp/CaffeProject/caffe/.build_release/tools/caffe.bin train --solver=voc12/config/VGG-16/solver_train.prototxt --gpu=0 --weights=voc12/model/VGG-16/init.caffemodel
I0202 23:35:38.493860 9563 caffe.cpp:185] Using GPUs 0
I0202 23:35:38.783365 9563 caffe.cpp:190] GPU 0: GeForce GTX 960M
I0202 23:35:40.776765 9563 solver.cpp:48] Initializing solver from parameters:
train_net: "voc12/config/VGG-16/train_train.prototxt"
base_lr: 0.001
display: 20
max_iter: 20000
lr_policy: "poly"
power: 0.9
momentum: 0.9
weight_decay: 0.0005
snapshot: 20000
snapshot_prefix: "voc12/model/VGG-16/train"
solver_mode: CPU
device_id: 0
average_loss: 20
I0202 23:35:40.776993 9563 solver.cpp:81] Creating training net from train_net file: voc12/config/VGG-16/train_train.prototxt
I0202 23:35:40.806784 9563 net.cpp:49] Initializing net from parameters:
name: "VGG-16"
state {
phase: TRAIN
}