Converting additional models from ConvNext repo to Hugging Face

by alexliao - opened Nov 15, 2022

Nov 15, 2022

Hi! I have been using the ConvNext models available on Hugging Face now with success, but I would like to try out some of the other pre-trained models (convnext-small-384-22k-1k for example) that are available on the ConvNext repo. What's the process for converting the .pth files to the .bin files that we see here?

nielsr

Nov 21, 2022

Hi,

Thanks for your interest in ConvNeXt :) oh it seems they released some new checkpoints which weren't available at the time I ported the model. We can extend this conversion script to port the new checkpoints

alexliao

Nov 22, 2022

Thanks for the pointer! I was able to get it working :)

alexliao changed discussion status to closed Nov 22, 2022

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment