Оптимізатори

Час дослідити різні оптимізатори, які ви можете використовувати для тренування моделі.

Для вас визначено користувацьку функцію train_model(optimizer, net, num_epochs). Вона приймає оптимізатор, модель і кількість епох як вхідні параметри, запускає цикли тренування та в кінці виводить тренувальну втрату.

Скористаймося train_model(), щоб запустити кілька коротких тренувань з різними оптимізаторами та порівняти результати!