Настройка оптимизатора
Теперь, когда логика обучения определена, нужно указать, как именно оптимизировать параметры модели.
В этом упражнении вы завершите реализацию метода configure_optimizers в модуле PyTorch Lightning, предназначенном для задач классификации изображений. Ваша цель — настроить оптимизатор, который будет обновлять параметры модели в процессе обучения. Для этого используйте оптимизатор Adam со скоростью обучения 1e-3.
Это упражнение является частью курса
Масштабируемые модели ИИ с PyTorch Lightning
Инструкции к упражнению
- Создайте оптимизатор Adam на основе параметров модели, задав скорость обучения равной
1e-3.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
import torch
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer