tinygrad/extra/models
George Hotz cbb8486779
ResNet training changes (update benchmark) (#2390)
* default arg for chunk

* bring back to_

* good changes

* new set

* unused hash

* fix optim

* new torch loader

* fix test lr scheduler
2023-11-22 17:41:12 -08:00
..
bert.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
convnext.py move fetch to helpers (#2363) 2023-11-19 12:29:51 -08:00
efficientnet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
mask_rcnn.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
resnet.py ResNet training changes (update benchmark) (#2390) 2023-11-22 17:41:12 -08:00
retinanet.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
rnnt.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
transformer.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
unet3d.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00
vit.py move things, clean up extra (#2292) 2023-11-13 20:18:40 -08:00