当前位置：网站首页>torch‘ has no attribute ‘inference_mode‘

torch‘ has no attribute ‘inference_mode‘

2022-07-27 10:59:00 【想搞钱的小陈】

在overstack flow上看到有人说是因为版本问题，（这个命令在torch.1.9以上），但下面有人反映说这个不行；

解决方案：

将代码替换，将torch.inference_mode()替换为torch.no_grad()

@torch.no_grad()
# @torch.inference_mode()
    with torch.no_grad():
    # with torch.inference_mode():

原网站

版权声明
本文为[想搞钱的小陈]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_34845880/article/details/121628253

边栏推荐

猜你喜欢

随机推荐