Files
tinygrad/extra/models/resnet.py
reddyn12 f1c7944c44 Fix batchnorm shapes for resnet.load_pretrained (#5167)
* Fix batchnorm shapes

* make it general reshape
2024-06-26 18:44:10 -04:00

6.2 KiB