Object Detection
YOLOv10
computer-vision
pypi

Leverage PyTorchModelHubMixin

#1
by nielsr HF staff - opened

Hi @kadirnar ,

Thanks so much for uploading YOLOv10 checkpoints to make them easier discoverable. :)

Small note though, we prefer to have each model checkpoint uploaded to a separate repository. With the mixin class, it allows to add from_pretrained and push_to_hub capabilities to each model, download stats, automatic model card, etc.

See the docs here: https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin

Opened a PR here: https://github.com/THU-MIG/yolov10/pull/168

Sign up or log in to comment