[ 组合替代实现 ]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()