在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():
在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():