nebuly-ai / optimate

A collection of libraries to optimise AI model performances
https://www.nebuly.com/
Apache License 2.0
8.37k stars 639 forks source link

Example using Pytorch dataloader with a Coco Detection dataset is needed #177

Open tobymcclean opened 1 year ago

tobymcclean commented 1 year ago

Unable to find an example that uses a Pytorch dataset and, more importantly, one that uses a Coco Detection dataset.

When running the Yolov8 example, the following error is the output:

[ERROR] Unexpected exception AssertionError('The dataloader must include label to measure the metric!') happened during tuning.

The error makes sense, but there is no obvious way to use a real Coco Detection dataset. See for example, the OpenVINO Notebook for Yolov8 : 230-yolov8-optimization.ipynb

diegofiori commented 1 year ago

Hello @tobymcclean! Thank you for reporting the error! Could you please share with us the code snippet that generates the error?