torch.set_deterministic_debug_mode

torch.set_deterministic_debug_mode(debug_mode)[源代码]

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

注意

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

参数

debug_mode (str 或 int) – 如果为 "default" 或 0，不对非确定性操作进行错误或警告。如果为 "warn" 或 1，在非确定性操作时发出警告。如果为 "error" 或 2，在遇到非确定性操作时抛出错误。