Jacky Lee 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