【发布时间】：2017-08-08 10:16:49
【问题描述】：
我正在尝试用我自己的数据训练一个模型，尽管它在几次迭代后开始训练它会抛出一个错误。这就是它的样子
0803 13:41:12.052456 3772 solver.cpp:237] Train net output #2:
loss = 1.82591 (* 1 = 1.82591 loss)
I0803 13:41:12.052477 3772 sgd_solver.cpp:105] Iteration 120, lr =
9.92e -05
F0803 13:41:18.020836 3779 data_transformer.cpp:168] Check failed:
height <= datum_height (227 vs. 224)
*** Check failure stack trace: ***
@ 0x7f8f7b14adaa (unknown)
@ 0x7f8f7b14ace4 (unknown)
@ 0x7f8f7b14a6e6 (unknown)
@ 0x7f8f7b14d687 (unknown)
@ 0x7f8f7b7ba9e0 caffe::DataTransformer<>::Transform()
@ 0x7f8f7b835a2f caffe::DataLayer<>::load_batch()
@ 0x7f8f7b8c6a5c
caffe::BasePrefetchingDataLayer<>::InternalThreadEntry()
@ 0x7f8f7b79aa30 caffe::InternalThread::entry()
@ 0x7f8f7b79b376 boost::detail::thread_data<>::run()
@ 0x7f8f7172aa4a (unknown)
@ 0x7f8f6bd73184 start_thread
@ 0x7f8f79a7dffd (unknown)
@ (nil) (unknown)
Aborted (core dumped)
我确实认为问题可能出在输入图像上，关于我应该更改或编辑的内容有什么建议吗？
【问题讨论】：
-
您的输入图像网络定义高度大于（227）比您的输入图像（224）。更改定义（train.prototxt）dim 2：从 227 到 224
-
感谢您的宝贵时间。我试试看
-
告诉我进展如何
-
抱歉回复晚了-现在我收到
I0806 09:41:02.010442 2992 sgd_solver.cpp:105] Iteration 360, lr = 9.76e-05 F0806 09:41:20.544955 2998 data_transformer.cpp:168] Check failed: height <= datum_height (224 vs. 199)任何关于我应该做什么的想法？ -
我认为您需要在 data_transformer.cpp 中查看失败的位置