mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-08 14:43:57 -05:00
readme updates
This commit is contained in:
@@ -12,6 +12,8 @@ This may not be the best deep learning framework, but it is a deep learning fram
|
||||
|
||||
The Tensor class is a wrapper around a numpy array, except it does Tensor things.
|
||||
|
||||
Update: tinygrad also somewhat supports GPUs through pyopencl. Our end goal is to train an ImageNet EfficientNet-B0.
|
||||
|
||||
### Installation
|
||||
|
||||
```bash
|
||||
@@ -76,6 +78,12 @@ loss.backward()
|
||||
optim.step()
|
||||
```
|
||||
|
||||
### ImageNet inference (on the micrograd puppy)
|
||||
|
||||
```bash
|
||||
python3 examples/examples/efficientnet.py
|
||||
```
|
||||
|
||||
### The promise of small
|
||||
|
||||
tinygrad will always be below 1000 lines. If it isn't, we will revert commits until tinygrad becomes smaller.
|
||||
|
||||
Reference in New Issue
Block a user