@Andrei1penguin1

Почему Keras не принимает функцию потерь mse?

Доброго времени суток, есть задача детектирования объекта на изображении, соответственно, модель заканчивается слоем
Dense(4, activation="sigmoid")
Компилируется следующим образом:
model.compile(optimizer="rmsprop", loss="mse")
Но при попытке обучить её выскакивает следующая ошибка:
ValueError: A target array with shape (25, 200, 200, 3) was passed for an output of shape (None, 4) while using as loss `mean_squared_error`. This loss expects targets to have the same shape as the output.
Подскажите пожалуйста, как это можно исправить?
  • Вопрос задан
  • 50 просмотров
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Войти через центр авторизации
Похожие вопросы