在使用PyTorch框架进行训练时,若需要输出结果排除随机性,则需要设置确定性计算开关。在开启确定性计算时,当使用相同的输入在相同的硬件和软件上执行相同的操作,输出的结果每次都是相同的。
[object Object]
确定性计算的用法和效果具体可参考相应官方文档,本小节仅介绍开启确定性计算的方法。
[object Object]
开启确定性计算开关:
[object Object]验证设置是否成功。
执行如下命令查询接口设置:
[object Object]回显示例如下:
[object Object]
执行训练时,打印此接口的返回值为True表示当前已开启确定性计算开关,返回False则表示未开启。