Allow execution on CPU, fix ImageNet head for timm models#216

Open

Daniel300000 wants to merge 2 commits intothuml:masterfrom

Daniel300000:master

Commits on May 4, 2023

Allow execution on CPU
Daniel300000
authored

Commits on May 25, 2023

Update utils.py, fix ImageNet head for timm models
Daniel300000
authored