Jacky LeeandGitHub d768acb6a9 Rename Normalize and move to nn (#415) ...

* Rename Normalize and move to nn

* Fix comparison to None error

* Add test for GroupNorm

* Rename test case

* Flip parameters to match PyTorch

* Increase error tolerance

* Fix elementwise_affine on channels

* Match arguments with PyTorch

* Initialize weight and bias only when affine is true

* Is this it?

* A bit cleaner

* Handle case where weight or bias is None

2023-01-25 07:47:59 -08:00

..

yolo

Fix yolo param name (#306)

2022-02-10 01:40:24 -07:00

benchmark_train_efficientnet.py

fix efficientnet

2022-09-28 14:23:01 -07:00

efficientnet.py

model touchups

2021-11-30 11:13:34 -05:00

mnist_gan.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

serious_mnist.py

Reshape dataset from fetch_mnist (#390)

2022-09-24 21:16:29 -04:00

stable_diffusion.py

Rename Normalize and move to nn (#415)

2023-01-25 07:47:59 -08:00

train_efficientnet.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

train_resnet.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

transformer.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

vgg7.py

tinygrad.nn (#367)

2022-08-18 07:41:00 -07:00

vit.py

support larger ViT models

2021-12-12 10:45:10 -08:00

yolov3.py

Correct spelling mistakes (#314)

2022-04-05 05:22:18 -07:00