Процесс создания модели в pytorch

• Класс nn содержит все необходимое для создания нейросети
• nn.Parameter - какие параметры может наша модель пробовать обучить, обычно слой PyTorch будет настраивать их
• nn.Module - базовый класс нейросетей, все модели потомки этого класса. Должен содержать переопределенную процедуру forward
• torch.optim - оптимизатор при создании модели
• torch.inference_mode() - прогнозирование