Почему потеря моей модели CNN продолжает расти при обучении набора данных знака трафика? ⇐ Python

[Anonymous]

Я обучаю модель CNN для классификации знаков трафика и сталкиваюсь с проблемой, в которой потеря продолжает увеличиваться, а не уменьшаться с каждой эпохой. /> Основные моменты: < /strong> < /p>

[*] Потеря обучения начинается около ~ 0,00307 и увеличивает < /li>
Потеря проверки также увеличивается. < /li>
Остается очень низкой. Настройка: [/b]

framework : pytorch

[*] dataSet : настраиваемые изображения знака трафика (~ 40 классов), включены в репо
: /> model : cnn (conv2d, maxpooling, falten, dense)

[*] Функция проигрыша : категорическая кроссентропия

[*]

[*]

[*]

[*] размер пакета : 32

[*] эпохи : 20

< /li>
переключенный оптимизатор от Adam на Sgd. < /p>
< /li>
< /ul>
ожидаемый результат: < /strong>
Потеря должна постепенно уменьшаться, поскольку модель учится, а точность должна улучшаться по сравнению с эпохами. Тем не менее, потеря продолжает расти. Я думаю, что я сделал ошибку класса TeptImagedAtaset (набор данных) или класс ImageDataset (набор данных), но я все еще смотрел все, что все еще не мог найти.
Я попытался снизить скорость обучения и увеличить количество эпох, но проблема с увеличением потери сохраняется. 2/20, потеря: 0,0029359486130283476,
epoch 3/20, потеря: 0,0028563185578263798,
epoch 4/20, потеря: 0,0030248229305273556,
epoch 5/20, проигрыш. :0.002645829956838785,
Epoch 6/20, Loss :0.002885574423275025,
Epoch 7/20, Loss :0.0028104307600957923,
Epoch 8/20, Loss :0.0026011793687331928,
Epoch 9/20, потеря: 0,0026105366173226137,
epoch 10/20, потеря: 0,0028680448236512125,
epoch 11/20, потеря: 0,0029950333832008648,
epoch 12/20, потеря :0.0027484399848427704,
Epoch 13/20, Loss :0.002788932249557719,
Epoch 14/20, Loss :0.002972953665703778,
Epoch 15/20, Loss :0.002865532483012213,
Epoch 16/20, потеря: 0,0028002487699125955,
Epoch 17/20, потеря: 0,002698509961512505,
Epoch 18/20, потеря: 0,002725444454547826,
epoch 19/20, проигрыш. : 0,003152967666140583,
Epoch 20/20, потеря: 0,002642994020346911

Подробнее здесь: https://stackoverflow.com/questions/797 ... sign-datas