Request for guidelines on training on multitask classification

Question

Request for guidelines on training on multitask classification

gd1925 opened this issue 10 months ago · comments

gd1925 commented 10 months ago

Hi,

I'm looking to train a custom dataset for multitask classification and I'm interested in trying to use the MiVOLO. Could you please provide some guidelines or advice on how to proceed with this? Any help would be greatly appreciated!

Thank you.

Maksim Kuprashevich · Answer 1 · Mon Sep 04 2023 02:36:35 GMT+0800 (China Standard Time)

Hello! Sorry for the delayed response, was on vacation.

You can definitely do it, and it's not very complicated. To achieve this, you'll need to integrate MiVOLO with the original timm train code (please note that we can't provide this specific code at the moment). The procedure is similar to working with regular models. However, in your case, there will be N outputs and N loss functions. You can combine them and experiment accordingly.