Add import to README

graphcore-research · Sep 28, 2023 · 573c811 · 573c811
1 parent 6255923
commit 573c811
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -7,6 +7,8 @@ pip install git+https://github.com/graphcore-research/pytorch-tensor-tracker
 ```
 
 ```python
+import tensor_tracker
+
 with tensor_tracker.track(module) as tracker:
     module(inputs).backward()