Open mrdbourke opened 1 year ago
See this notebook for an example conversion: https://github.com/mrdbourke/nutrify/blob/main/foodvision/notebooks/06_export_model.ipynb
Note: Prediction/evaluation can only take place on macOS with CoreML, should setup some code to make a prediction over ~100 images with CoreML & PyTorch for a sanity check on how the model is performing.
Could have multiple export types as well:
More:
python export.py
for a model of multiple kinds to be created somewhere