estenhl/pyment-public

Unable to find pre-trained weights

amirro1 opened this issue · 2 comments

amirro1 commented

Hi,

Thanks for this project. You mention in the readme that users should be able to run a pre-trained model by downloading existing weights. However, I can't find an example of how these weights are downloaded in the notebooks. Is there an h5 file (or similar) somewhere with the weights that I could access?

Thanks

estenhl commented

Sorry for the late reply, there are some google drive urls in the weight repo you can use if you want to download and inject them manually:

https://github.com/estenhl/pyment-public/blob/main/pyment/models/utils/weight_repository.py

I would generally recommend using the built-in functionality when possible though (e.g. weights=... as a parameter to the model class)

amirro1 commented

Thanks so much! I totally missed that file when I was looking through the repo. I will plan on using the built-in functionality, I just wasn't sure where those weights were being sourced from.

Thanks again!