diff --git a/micrograd/engine.py b/micrograd/engine.py index 654f443b..9fe5b2f0 100644 --- 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: