oneflow.nn.Module.parameters¶
-
Module.
parameters
(recurse=True) → Iterator[Parameter]¶ Returns an iterator over module parameters.
This is typically passed to an optimizer.
- Parameters
recurse (bool) – if True, then yields parameters of this module and all submodules. Otherwise, yields only parameters that are direct members of this module.
- Yields
Parameter – module parameter
Example:
>>> for param in model.parameters(): ... print(type(param), param.size()) <class 'oneflow.Tensor'> oneflow.Size([10])