Как я могу изменить размер мини-партии при обучении в маске R-CNN для обнаружения объектов и сегментации?
Я пытаюсь обучить модели маскируемого RCNN ( https://github.com/matterport/Mask_RCNN) для обнаружения номерного знака. Я следовал экзамену по обнаружению баллона и тренировался без изменения каких-либо гиперпараметров (за исключением многопроцессорности = false, поскольку я использую CPU), наблюдая за тренировкой, я заметил, что функция потерь начиналась высоко в первые эпохи и много раз поднималась и опускалась от одного шага к другому, и к 23-й эпохе (только из 30) она опустилась только до 0,70 с выше 1 значения val_loss, так что я решил, что должен увеличить размер пакета (оставленный размер пакета =1), так как мой набор данных изображения довольно разнообразен, чтобы уменьшить дисперсию функции потерь, но я не могу делать это каждый раз Я изменяю размер пакета в config.py и запускаю тренировку, она показывает мне РАЗМЕР БАТАРЕИ = 1 каждый раз, я в настоящее время перезапускаю тренировку с весами, сгенерированными в 23-й эпохе.
это результат, который я получаю после переобучения модели с весами, сгенерированными в 23-й эпохе:
Я новичок в нейронных сетях, так что не обращайте на меня внимания, пожалуйста, любая помощь будет очень полезна, всем большое спасибо и хорошего дня.