torch.is_inference_mode_enabled

torch.is_inference_mode_enabled()

如果当前开启了推理模式,返回 True。

本页目录