Я пытаюсь взять сеть, подготовленную для классификации imagenet, и настроить ее для сегментации изображения.Я удалил полностью связанный слой и заменил его сверткой 1x1, чтобы сгенерировать маску для каждого класса.Я также заменил слой потерь на потери шарнира L1.
На итерации 0 потеря составляет около 10000, а затем на итерации 20 она становится равной 0. Когда она достигает тестовой итерации, она выдает выходной сигнал Test net -nan.
Есть мысли, где это может пойти не так?