av-savchenko/face-emotion-recognition

Request for guidelines on training on multitask classification

gd1925 opened this issue · 2 comments

gd1925 commented

Hi,

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

Thank you.

av-savchenko commented

Hello! Please look at the section "Multi-task: FER+Valence-Arousal" at train_emotions-pytorch.ipynb. Section "Multi-task challenge" at abaw3_train.ipynb and abaw4_mtl_train.ipynb may also help, though in the latter notebook, I actually train three heads separately

  • 😄1
gd1925 commented

Hi @HSE-asavchenko, Thanks a ton for your reply! I will try to see whichever I can understand and try to train using it.