Is it safe to push trained model weight to GitHub?
Hi @jabertuhin, this repo of yours is really useful for me to learn to deploy a deep learning project using Docker and FastAPI. I just want to know if it is a good practice to push the .pth file containing the model weight onto GitHub? Thank you in advance!
Hello @XuanVuNguyen, it's great to hear that this repository helped you to learn. And about the question - it's one of the way to serve ml model. For small projects or MVP projects, I think it's sufficient to meet the goal, hence the simple approach. Though it comes with some pros and cons.
Obviously it will be good to save the model weights in a different storage(ex: aws S3 bucket) and download it while running the application.
But at the end end it depends on the use case. Sharing a blog post which might help you to understand different approaches of serving models in production - How to Serve Models
Thank you @jabertuhin for the materials. Much appreciate!