ComenzarEmpieza gratis

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

Ver curso

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 = ____
Editar y ejecutar código