ComenzarEmpieza gratis

Optimizadores

Es hora de explorar los distintos optimizadores que puedes utilizar para entrenar tu modelo.

Se ha definido para ti una función personalizada llamada train_model(optimizer, net, num_epochs). Toma el optimizador, el modelo y el número de épocas como entradas, ejecuta los bucles de entrenamiento e imprime la pérdida de entrenamiento al final.

Utilicemos train_model() para realizar unos cuantos entrenamientos cortos con diferentes optimizadores y comparemos los resultados.

Este ejercicio forma parte del curso

Aprendizaje profundo intermedio con PyTorch

Ver curso

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

import torch.optim as optim

net = Net()

# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)

train_model(
    optimizer=optimizer,
    net=net,
    num_epochs=10,
)
Editar y ejecutar código