github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-08 14:43:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

more graph docs

Browse Source
This commit is contained in:
George Hotz
2022-06-05 12:16:50 -07:00
parent 88de42fb6e
commit 89acf6742d
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
README.md
View File

@@ -163,6 +163,13 @@ See `examples/yolov3.py`
tinygrad will always be below 1000 lines. If it isn't, we will revert commits until tinygrad becomes smaller.
### Drawing Execution Graph
* Nodes are Tensors
* Black edge is a forward pass
* Blue edge is a backward pass
* Red edge is data the backward pass depends on
* Purple edge is intermediates created in the forward
```bash
GRAPH=1 python3 test/test_mnist.py TestMNIST.test_conv_onestep
dot -Tsvg /tmp/net.dot -o /tmp/net.svg