[ 组合替代实现 ]torch.inference_mode¶

torch.inference_mode ¶

          torch.inference_mode(mode=True)

         

Paddle 无此 API，需要组合实现。

转写示例¶

          # PyTorch 写法
@torch.inference_mode(False)

# Paddle 写法
# 当 mode 为 False 时，可以直接删去

# PyTorch 写法
@torch.inference_mode(True)

# Paddle 写法
@paddle.no_grad()