Hey @usmanshahid @milLII3yway ,
Have a look at our new task framework: GitHub - Lightning-AI/lightning-flash: Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains.
We would love for you to contribute new tasks.
Best,
T.C