search

FrancescoSaverioZuppichini / glasses

High-quality Neural Networks for Computer Vision 😎
https://francescosaveriozuppichini.github.io/glasses-webapp/
MIT License
431 stars 37 forks source link

Feature/#258 #259

Closed FrancescoSaverioZuppichini closed 2 years ago

FrancescoSaverioZuppichini commented 3 years ago

Added custom providers, now we can

from glasses.utils.weights import LocalPretrainedWeightsProvider, HFPretrainedWeightsProvider

AutoModel.from_pretrained('resnet18', provider=LocalPretrainedWeightsProvider(root='.')) # using local weights
AutoModel.from_pretrained('resnet18', provider=HFPretrainedWeightsProvider()) # using hugging face hubs - default