mindspore.set_deterministic

查看源文件
mindspore.set_deterministic(deterministic)[源代码]

是否开启确定性计算。

此配置为全局配置,开启后,后续计算操作遵从配置要求。当开启确定性计算功能时,算子在相同的硬件和输入下,多次执行将产生相同的输出。但启用确定性计算往往导致算子执行变慢。

框架默认不开启确定性计算。

说明

  • 在分布式场景下,建议用户在调用接口 mindspore.communication.init() 前设置确定性计算,以保证使能全局通信域上的通信算子确定性。

  • 确定性计算固定方法必须与待固定的网络、算子等在同一个主进程。同一线程中只能设置一次确定性状态,暂不推荐在一个线程中多次设置确定性状态。

参数:
  • deterministic (bool) - 是否开启确定性计算。

样例:

>>> import mindspore as ms
>>> ms.set_deterministic(True)