Fix train_efficientnet
MODIFY
tinygrad/tensor.py
tinygrad/tensor.py
@@ -120,7 +120,7 @@ class Tensor:
visited, nodes = set(), []
def deepwalk(node):
- visited.add(self)
+ visited.add(node)
if node._ctx:
for i in node._ctx.parents:
if i not in visited:

Fix train_efficientnet (#120)

Co-authored-by: holonomicjl <58403584+holonomicjl@users.noreply.github.com>