George Hotz 7276f8d6bf improve constant folding, detach before moving tensor

2022-07-02 15:29:40 -07:00

..

augment.py

minor optimizations & cleaning (#257)

2021-06-02 09:57:15 -07:00

gradcheck.py

move gradcheck to extra, clean up unbroadcast

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

onnx.py

comments

2022-06-26 17:20:25 -07:00

training.py

improve constant folding, detach before moving tensor

2022-07-02 15:29:40 -07:00

utils.py

don't track_running_stats, parameters must require_grad

2022-07-02 14:38:45 -07:00