Modules — PyTorch 1.10.1 documentation
pytorch.org › docs › stableModules make it simple to specify learnable parameters for PyTorch’s Optimizers to update. Easy to work with and transform. Modules are straightforward to save and restore, transfer between CPU / GPU / TPU devices, prune, quantize, and more. This note describes modules, and is intended for all PyTorch users.
ModuleList — PyTorch 1.10.1 documentation
pytorch.org › generated › torchModuleList. Holds submodules in a list. ModuleList can be indexed like a regular Python list, but modules it contains are properly registered, and will be visible by all Module methods. Appends a given module to the end of the list. Appends modules from a Python iterable to the end of the list.