Спасибо, подзабыл, что такая опция есть. А еще, есть ли гайд, как в процессе обучения регулировать, например, learning_rate?
А так, пары правил достаточно:
- если loss дико скачет с самого начала, растет там и т.д., то lr слишком большой
- если loss падал на постоянном lr и перестал, то стоит попробовать уменьшить lr, например в 3 или в 10 раз, и так пока модель точно не перестанет учится т.е. оптимум примерно достигнут