А как насчет потери тренировок? Обратите внимание, что total_loss
- это потеря проверки здесь.
Если ваша потеря на тренировке уменьшается, а потеря на валидацию увеличивается, это явно признак переобучения, вы можете использовать потерю регуляризации во время тренировки, добавив следующее в файл конфигурации, в части train_config
add_regularization_loss: true
так же, как batch_size: 24