torch.set_deterministic_debug_mode

torch.set_deterministic_debug_mode(debug_mode)

设置确定性操作的调试模式。

注意

这是一个替代接口，用于 torch.use_deterministic_algorithms()。有关受影响操作的详细信息，请参阅该函数的文档。

Parameters

debug_mode (str 或 int) – 如果为“default”或0，不在非确定性操作上出错或警告。如果为“warn”或1，在非确定性操作上警告。如果为“error”或2，在非确定性操作上出错。