Update recipes_source/compiling_optimizer.py

malfet · web-flow · commit 16fd39ab4fd4 · 2024-01-10T14:56:12.000-08:00
diff --git a/recipes_source/compiling_optimizer.py b/recipes_source/compiling_optimizer.py
@@ -59,7 +59,7 @@
 opt = torch.optim.Adam(model.parameters(), lr=0.01)
 
 
-@torch.compile()
+@torch.compile(fullgraph=False)
 def fn():
     opt.step()
 
