Recommended unit tests for CV model training pipelines?

If you were tasked with automating a CV model training pipeline, what are the set of unit-tests that you would run before model deployment?

Would defining tests for the model’s performance and robustness be enough?

What tests do you usually define?

Do they change based on the task at hand (detection, classification, segmentation, etc)?

5 Likes