mindspore.parallel.strategy.get_current_strategy_metadata

View Source On Gitee
mindspore.parallel.strategy.get_current_strategy_metadata(network)[source]

Get parameters dictionary of cur rank of the network.

Parameters

network (str) – The network name.

Returns

Dict. The key is 0 (representing the local rank), and the value is the slicing strategy for all parameters. The key within the value represents the parameter name, and the value is the corresponding slicing strategy for that parameter. If not supported, returns None.