Fix torch_compile_tutorial.py (#2496)

williamwen42 · web-flow · commit 22d878208f4e · 2023-06-29T13:18:40.000-07:00
add `torch.no_grad` call to `evaluate`
diff --git a/intermediate_source/torch_compile_tutorial.py b/intermediate_source/torch_compile_tutorial.py
@@ -142,7 +142,8 @@ def init_model():
 # ``mode`` argument, which we will discuss below.
 
 def evaluate(mod, inp):
-    return mod(inp)
+    with torch.no_grad():
+        return mod(inp)
 
 model = init_model()
 
