با عرض سلام و خسته نباشید
من داده هام رو روی شبکه alexnet آموزش دادم اما loss تغییر چندانی نمیکنه و کم نمیشه . مشخصات solver به صورت زیر تنظیم کردم :
چون تعداد تصاویر آموزشم 661 تصویر بود ، بچ سایز مرحله ی آموزش و 60 گرفتم تا بعد از 110 تکرار حدودا 10 ایپاک تکرار شده باشه .
همانطور که آقای حسن پور توی سایت گفتن ، حاصلضرب عدد بچ در مرحله ی تست و عدد test_iter باید برابر با تعداد تصاویر تست باشه ، برای همین بچ سایز در مرحله ی تست رو 8 و تعداد itest_iter را 28 گرفتم چون 224 تصویر تست داشتم .
base-lr ابتدا 0.01 بود اما چون loss زیاد کم نمیشد روی 0.001 گذاشتم و به همین خاطر مقدار momentum=099 انتخاب کردم . اما باز هم loss از 2 کمتر نمیشه و دقت هم حداکثر 0.098 میشه . ممنون میشم راهنماییم کنین.
این مربوط به آموزش شبکه از اول بود . اگه finetune کنم و فقط لایه ی آخر trainبشه اول loss کم میشه اما بعد از 10 تکرار زیاد میشه و به nan میرسه
فایل solver :
net: "/home/amir/Develop/caffe/models/my_corel_caffenet/train_val.prototxt"
test_iter: 28
test_interval: 110
base_lr: 0.01
lr_policy: "step"
gamma: 0.1
stepsize: 100000
display: 1
max_iter: 115
momentum: 0.9
weight_decay: 0.0005
snapshot: 100
snapshot_prefix: "/home/amir/Develop/caffe/models/my_corel_caffenet/caffenet_train"
solver_mode: CPU