Marcel BischoffandGitHub dc8fa7999c Transpose on GPU (#221) ...

* 2serious

* load/save

* fixing GPU

* added DEBUG

* needs BatchNorm or doesn't learn anything

* old file not needed

* added conv biases

* added extra/training.py and checkpoint

* assert in test only

* save

* padding

* num_classes

* checkpoint

* checkpoints for padding

* training was broken

* merge

* rotation augmentation

* more aug

* needs testing

* streamline augment, augment is fast thus bicubic

* tidying up

* transformer eval

* axis=-1

* transpose

* test for permutation using torch.movedims

* another test

* line

2020-12-29 10:40:11 -05:00

..

augment.py

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

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

efficientnet.py

issue-193 - Move torch loader out of efficientnet code (#213)

2020-12-22 00:19:16 -05:00

gradcheck.py

move gradcheck to extra, clean up unbroadcast

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

training.py

Transpose on GPU (#221)

2020-12-29 10:40:11 -05:00

utils.py

issue-193 - Move torch loader out of efficientnet code (#213)

2020-12-22 00:19:16 -05:00