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
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,
)