Markus Enzweiler
|
2b61d9deb6
|
Updated CIFAR-10 ResNet example to use BatchNorm instead of LayerNorm (#257)
* replaced nn.LayerNorm by nn.BatchNorm
* mlx>=0.0.8 required
* updated default to 30 epochs instead of 100
* updated README after adding BatchNorm
* requires mlx>=0.0.9
* updated README.md with results for mlx-0.0.9
|
2024-01-12 05:43:11 -08:00 |
|
Awni Hannun
|
27c0a8c002
|
Add llms subdir + update README (#145)
* add llms subdir + update README
* nits
* use same pre-commit as mlx
* update readmes a bit
* format
|
2023-12-20 10:22:25 -08:00 |
|
Awni Hannun
|
b1b9b11801
|
updates + format
|
2023-12-14 12:09:10 -08:00 |
|
Sarthak Yadav
|
29b7a97342
|
updated header
|
2023-12-14 16:28:00 +01:00 |
|
Sarthak Yadav
|
15a6c155a8
|
simplified ResNet, expanded README with throughput and performance
|
2023-12-14 09:05:04 +01:00 |
|
Sarthak Yadav
|
2439333a57
|
fixed doc for ResNet
|
2023-12-12 19:07:39 +01:00 |
|
Sarthak Yadav
|
f37e777243
|
added CIFAR10 + ResNet example
|
2023-12-12 19:01:06 +01:00 |
|