mindspore.parallel.strategy.get_current_strategy_metadata

View Source On AtomGit
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.