Actualización manual de los pesos
Ahora que sabes cómo acceder a los pesos y sesgos, realizarás manualmente el trabajo del optimizador PyTorch. Aunque PyTorch lo automatiza, practicarlo manualmente te ayuda a adquirir intuición sobre cómo aprenden y se ajustan los modelos. Esta comprensión será valiosa a la hora de depurar o ajustar las redes neuronales.
Se ha creado una red neuronal de tres capas y se ha almacenado como variable model
. Esta red se ha utilizado para un pase hacia adelante y se han calculado la pérdida y sus derivadas. Se ha elegido una tasa de aprendizaje predeterminada, lr
, para escalar los gradientes al realizar la actualización.
Este ejercicio forma parte del curso
Introducción al aprendizaje profundo con PyTorch
Ejercicio interactivo práctico
Prueba este ejercicio completando el código de muestra.
weight0 = model[0].weight
weight1 = model[1].weight
weight2 = model[2].weight
# Access the gradients of the weight of each linear layer
grads0 = ____
grads1 = ____
grads2 = ____