torch.inference_mode(mode=True)
Paddle 无此 API,需要组合实现。
# PyTorch 写法 @torch.inference_mode(False) # Paddle 写法 # 当 mode 为 False 时,可以直接删去 # PyTorch 写法 @torch.inference_mode(True) # Paddle 写法 @paddle.no_grad()