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 y completa 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 = ____