From 9875b3e83987a1b7b6eea69d0bbc5a9074cda92e Mon Sep 17 00:00:00 2001 From: Douglas Orr Date: Thu, 28 Sep 2023 14:58:00 +0100 Subject: [PATCH] Sync README with module doc --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index e692a5e..4908f1d 100644 --- 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()) ```