search

huggingface / pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more
https://huggingface.co/docs/timm
Apache License 2.0
31.95k stars 4.73k forks source link

RuntimeError: Unknown model (coatnext_nano_rw_224.sw_in1k) #1733

Closed WikiGenius closed 1 year ago

WikiGenius commented 1 year ago

Hello

After installing timm I have this bug on google collab

bug1

Example code https://huggingface.co/timm/coatnext_nano_rw_224.sw_in1k

I want to test the pre-trained model COATNET because it is the latest of the art and I want to fine-tune it to be a multi-task classifier

Thanks so much

WikiGenius commented 1 year ago

It has been solved

pip install --pre timm

or

pip install git+https://github.com/rwightman/pytorch-image-models.git

Thanks so much