Atualização manual dos pesos
Agora que você sabe como acessar pesos e tendências, você executará manualmente o trabalho do otimizador do PyTorch. Embora o PyTorch automatize isso, praticar manualmente ajuda você a desenvolver a intuição de como os modelos aprendem e se ajustam. Esse entendimento será valioso para a depuração ou o ajuste fino das redes neurais.
Uma rede neural de três camadas foi criada e armazenada como a variável model
. Essa rede foi usada em uma passagem direta (forward pass), e a perda e seus derivados foram calculados. Uma taxa de aprendizado padrão, lr
(abreviação de “learning rate”), foi escolhida para dimensionar os gradientes ao fazer a atualização.
Este exercício faz parte do curso
Introdução ao Aprendizado Profundo com o PyTorch
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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 = ____