tinygrad/examples
Marcel Bischoff d204f09316
some progress on batchnorms (draft) (#147)
* no of categories for efficientnet

* need layer_init_uniforn

* merge fail

* merge fail

* batchnorms

* needs work

* needs work how determine training

* pow

* needs work

* reshape was needed

* sum with axis

* sum with axis and tests

* broken

* works again

* clean up

* Update test_ops.py

* using sum

* don't always update running_stats

* space

* self

* default return running_stats

* passes test

* need to use mean

* merge

* testing

* fixing pow

* test_ops had a line dropped

* undo pow

* rebase
2020-12-09 22:14:27 -08:00
..
efficientnet.py batchnorm work 2020-12-06 14:40:07 -08:00
serious_mnist.py add conv biases and max pool 2020-12-09 08:01:20 -08:00
train_efficientnet.py some progress on batchnorms (draft) (#147) 2020-12-09 22:14:27 -08:00