Використання оптимізатора PyTorch

Раніше ви вручну оновлювали вагу мережі, щоб краще зрозуміти, як відбувається навчання „під капотом". Проте цей підхід не масштабується для глибоких мереж із багатьма шарами.

На щастя, PyTorch надає оптимізатор SGD, який автоматизує цей процес усього кількома рядками коду. Тепер ви завершите цикл навчання, оновивши ваги за допомогою оптимізатора PyTorch.

Нейронну мережу створено й надано у змінній model. Цю модель використано для прямого проходу, щоб отримати тензор передбачень pred. One-hot закодований тензор називається target, а функцію втрат крос-ентропії збережено як criterion.

torch.optim як optim і torch.nn як nn уже завантажено для вас.