transfer-learning-conv-ai
transfer-learning-conv-ai copied to clipboard
Is it possible to do the auxilary task multiple times for each instance?
Hi,
Is it possible to define multiple mc_token for each instance and do an auxilary classification for each of them? If yes, can you guide me? Thanks
@thomwolf Will you please guide if possible? Many thanks.
Hi, Hannabrahman.
I think you can edit the functions build_input_from_segments
, and MODEL_INPUTS
in file train.py
.
And add your classification in the function update
Thanks @ZHANG45. But the GPT2DoubleHeadsModel
only accommodates two losses, right? LM and 1 auxiliary multiple choice loss. Should I change the `GPT2DoubleHeadsModel' class as well?