Fix caching - Githubissues

drivendataorg / zamba

A Python package for identifying 42 kinds of animals, training custom models, and estimating distance from camera trap videos

https://zamba.drivendata.org/docs/stable/

MIT License

118 stars 27 forks source link

Fix caching #190

Closed ejm714 closed 2 years ago

ejm714 commented 2 years ago

Main changes:

downloads public checkpoint file to top level model_cache_dir (this supports look ups in the correct place on future runs)
if cached checkpoint file exists (and no other checkpoint was passed in), put the cached path on the checkpoint property in TrainConfig

These changes allow the previously downloaded file to actually get used on future runs. Previously, we were seeing "downloading model weights" on every run.

Closes #189

github-actions[bot] commented 2 years ago

🚀 Deployed on https://deploy-preview-190--silly-keller-664934.netlify.app

ejm714 commented 2 years ago

All tests are now passing, this is ready for a review