Optionally use pre-computed topo

tekknolagi · Nov 23, 2024 · d2a2832 · d2a2832
1 parent dee784c
commit d2a2832
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/micrograd/engine.py b/micrograd/engine.py
@@ -77,7 +77,9 @@ def build_topo(v):
         build_topo(self)
         return topo
 
-    def backward(self, reverse_topo):
+    def backward(self, reverse_topo=None):
+        if reverse_topo is None:
+            reverse_topo = self.topo()[::-1]
         # go one variable at a time and apply the chain rule to get its gradient
         self.grad = 1
         for v in reverse_topo: