Module — PyTorch 1.10.1 documentation
pytorch.org › docs › stableregister_buffer (name, tensor, persistent = True) [source] ¶ Adds a buffer to the module. This is typically used to register a buffer that should not to be considered a model parameter. For example, BatchNorm’s running_mean is not a parameter, but is part of the module’s state. Buffers, by default, are persistent and will be saved ...
Parameter — PyTorch 1.10.1 documentation
pytorch.org › torchParameter¶ class torch.nn.parameter. Parameter (data = None, requires_grad = True) [source] ¶. A kind of Tensor that is to be considered a module parameter. Parameters are Tensor subclasses, that have a very special property when used with Module s - when they’re assigned as Module attributes they are automatically added to the list of its parameters, and will appear e.g. in parameters ...