2016-12-06 5 views
0

Я тренирую простую нейронную сеть с Caffe, из примера MNIST. Мне нужны две вещи во время тренировки:Как получить ошибку обучения во время обучения с помощью Caffe?

  1. ошибка классификации на обучающей выборки по 600 итераций
  2. Ошибка классификации на тестовом наборе в конце, после 30000 итераций

Возможно ли это?

Я узнал, как это сделать 1), заменив тестовый набор тренировочным комплектом и используя test_iter, но он кажется действительно взломан, и тогда у меня не может быть 2). Есть ли простой способ сказать Caffe вычислить ошибку обучения для каждой итерации display?

+0

Ошибка обучения, которую вы видите, является средней на мини-барах, а не всем набором. – Shai

+0

Да, но мне нужна ошибка для всего набора данных после каждого 600 итераций. –

ответ

1
  1. Может быть, вы можете добавить этапа тестирования в файле .prototxt и использовать Accuracy layer для вывода точности.
  2. Использование test_interval: 600 для установки интервала проверки.
+0

Я добавил вторую фазу, и теперь у меня есть точность поезда во время тренировки, но проблема заключается не в том, что у меня есть точность теста и точности на каждые 600 итераций. Но мне кажется, что это лучшее, что я могу сделать. –

+0

Возможно, это лучшее, что можно сделать в Caffe с несколькими этапами тестирования, поэтому я выбрал этот ответ. –

Смежные вопросы