Sync README with module doc

graphcore-research · Sep 28, 2023 · 9875b3e · 9875b3e
1 parent 58116a2
commit 9875b3e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -11,6 +11,8 @@ with tensor_tracker.track(module) as tracker:
     module(inputs).backward()
 
 print(list(tracker))
+# => [Stash(name="0.linear", type=nn.Linear, grad=False, value=tensor(...)),
+#     ...]
 
 display(tracker.to_frame())
 ```