Accessing and Modifying Model Parameters and Layers
Was this section helpful?
torch.nn.Module, PyTorch Authors, 2023 - The official documentation for the base class of all neural network modules in PyTorch, describing methods for accessing, inspecting, and managing submodules, parameters, and their states.
Deep Learning with PyTorch, Eli Stevens, Luca Antiga, and Thomas Viehmann, 2020 (Manning Publications) - A comprehensive textbook providing thorough explanations and practical examples of building, inspecting, and customizing neural networks using PyTorch, covering core concepts like nn.Module and parameter handling.
Transfer Learning for Computer Vision Tutorial, Sasank Chilamkurthy, 2017 - An official PyTorch tutorial demonstrating practical techniques for fine-tuning pretrained models, including freezing layers and replacing output layers, which involves direct manipulation of model parameters and architecture.