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])