COVID-19
COVID-19 copied to clipboard
Get Pytorch models ready for production
We want to use Pytorch models as lambda functions using AWS.
However, we need to reduce the size of the model. One way of doing that is using caffe and ONIX.
TO-DO:
- Create a script to convert a pytorch model to caffe using ONIX.
- Right documentation on how to use the script, pytorch version need, etc.
- Find if there is any alternative to reduce the size of a pytorch model for production