commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

readme

This commit is contained in:
George Hotz 2020-11-07 11:41:27 -08:00
parent 9b8fefe4e3
commit 5486135f2d
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -78,7 +78,7 @@ optim.step()
### GPU Support?!
tinygrad supports GPUs through PyOpenCL. Not all ops are supported yet.
tinygrad supports GPUs through PyOpenCL. Not all ops are supported yet on the backward pass.
```python
from tinygrad.tensor import Tensor
@ -93,6 +93,8 @@ Despite being tiny, tinygrad supports the full EfficientNet. Pass in a picture t
python3 examples/efficientnet.py https://upload.wikimedia.org/wikipedia/commons/4/41/Chicken.jpg
```
protip: Set "GPU=1" environment variable if you want this to go faster.
### The promise of small
tinygrad will always be below 1000 lines. If it isn't, we will revert commits until tinygrad becomes smaller.