Skip to content

pytorch-ingredient-detection-1.0

Compare
Choose a tag to compare
@raphael0202 raphael0202 released this 16 Aug 15:02
8bba859

This ingredient detection model was trained on the ingredient detection dataset v1 using code in this version of the repository.
Training was tracked on Wandb.

More information on experiments performed on can be found in this document.
This release provides the following assets:

Training-related assets:

  • predictions.tar.gz: predictions on train and test dataset of the model, in:
    • HTML format: easier to view
    • JSONL format: either the raw or the aggregated (post-processed) version
  • model-huggingface.tar.gz: the HuggingFace serialized model

Serving assets:

  • onnx.tar.gz: the model exported to ONNX format