Валидация потерь для пироха Faster-RCNN - PullRequest
1 голос
/ 21 февраля 2020

В настоящее время я выполняю обнаружение объектов в пользовательском наборе данных, используя обучение переносу из модели Faster-RCNN, прошедшей предварительную подготовку к pytorch (как в torchvision tutorial ). Я хотел бы вычислить dict потери проверки (как в режиме поезда) в конце каждой эпохи. Я могу просто запустить модель в режиме поезда для проверки: eval / train mode). И в режиме eval модель возвращает предсказанные bboxes (как и ожидалось). Могу ли я использовать некоторые встроенные функции для этого?

Спасибо.

...