Calculating Gradients and Updating Weights in PyTorch
Was this section helpful?
Autograd: automatic differentiation, PyTorch Developers, 2024 - Official documentation for PyTorch's automatic differentiation engine, explaining the computation graph, gradient calculation, and requires_grad attribute.
torch.optim, PyTorch Developers, 2024 (PyTorch Foundation) - Official documentation covering PyTorch's optimization packages, detailing the functions for zeroing gradients and updating parameters according to various algorithms.
Dive into Deep Learning, Aston Zhang, Zack C. Lipton, Mu Li, Alex J. Smola, 2023 (Cambridge University Press) - A comprehensive open-source textbook providing practical examples of training neural networks in PyTorch, including gradient calculation and weight updates.