George Hotz b80cacb416 fix GPU efficientnet example

2021-05-26 17:29:35 -07:00

..

augment.py

Big MNIST model with PIL augmentation and load/save (#160)

2020-12-13 20:45:55 -08:00

efficientnet.py

fix GPU efficientnet example

2021-05-26 17:29:35 -07:00

gradcheck.py

move gradcheck to extra, clean up unbroadcast

2020-11-16 08:03:31 -08:00

kinne.py

vgg7 (image upscaling) implementation - not the best, but it works (#255)

2021-05-12 23:48:51 -07:00

training.py

transformer >99.98% test accuracy in ~30s (#230)

2021-01-02 07:45:09 -08:00

transformer.py

if you like your transformers twice as slow, use the GPU

2020-12-29 17:14:23 -05:00

utils.py

remove print

2021-01-02 12:53:30 -08:00

waifu2x.py

vgg7 (image upscaling) implementation - not the best, but it works (#255)

2021-05-12 23:48:51 -07:00