named_buffers
- UNet.named_buffers(prefix: str = '', recurse: bool = True) Iterator[Tuple[str, torch.Tensor]]
Returns an iterator over module buffers, yielding both the name of the buffer as well as the buffer itself.
- Parameters
- Yields
(string, torch.Tensor) – Tuple containing the name and buffer
Example:
>>> for name, buf in self.named_buffers(): >>> if name in ['running_var']: >>> print(buf.size())