🚀 auto infer task type in `AutoClassification` from datamodule
aniketmaurya opened this issue · 3 comments
aniketmaurya commented
Is your feature request related to a problem? Please describe.
For image and text classification we have to do use ImageClassification or TextClassification.
This can be auto-inferred from the data type.
Describe the solution you'd like
automodel = AutoClassification(datamodule=dm, task='infer') # other task could be 'image' or 'text'
automodel.hp_tune()Kanter666 commented
Hi, I will look into this issue. I will let know today/tomorrow if I get stuck or need help.
aniketmaurya commented
awesome @Kanter666! feel free to ping me if you want to discuss 😄
github-actions commented
Stale issue message