tinygrad/models
Jacky Lee cb679cd051
Fix weight initialization (#566)
* Fix weight initialization

* Use scaled_uniform in serious_mnist
2023-02-19 11:25:29 -08:00
..
efficientnet.py BatchNorm2D -> BatchNorm2d (#558) 2023-02-16 12:31:49 -08:00
resnet.py BatchNorm2D -> BatchNorm2d (#558) 2023-02-16 12:31:49 -08:00
transformer.py Fix weight initialization (#566) 2023-02-19 11:25:29 -08:00
vit.py support larger ViT models 2021-12-12 10:45:10 -08:00