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 completando 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