tinygrad/extra/models
chenyu fcf4a5ccf2
fix example that calls Tensor.__bool__ (#3650)
also removed `.cpu()` calls in mask_rcnn so `python3 examples/mlperf/model_spec.py` runs
2024-03-07 16:59:26 -05:00
..
bert.py multilazybuffer fix (#3609) 2024-03-04 17:36:23 -05:00
convnext.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
efficientnet.py ban __bool__ on Tensor (#3632) 2024-03-06 17:12:35 -05:00
llama.py sharding for llama (#3151) 2024-01-16 19:28:00 -08:00
mask_rcnn.py fix example that calls Tensor.__bool__ (#3650) 2024-03-07 16:59:26 -05:00
resnet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
retinanet.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
rnnt.py move graph.py and jit.py into features (#3376) 2024-02-12 17:34:34 +01:00
transformer.py replace with tensor op (#3099) 2024-01-12 14:13:40 -05:00
unet3d.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00
vit.py move to new cached fetch (#2493) 2023-11-28 17:36:55 -08:00