BERT4doc-Classification icon indicating copy to clipboard operation
BERT4doc-Classification copied to clipboard

How to fine tuning model on multi-tasks?

Open swufelyc opened this issue 4 years ago • 3 comments

Sorry to bother you! But it seems to me, the run_classifier_single_layer.py does not save the model, and what should I do to fine tuning the fine tuned model? Thanks!

swufelyc avatar May 30 '20 13:05 swufelyc

Thank you for your issue! I have just uploaded our codes about the fine-tuning model on multi-tasks. The multi-task fine-tuning is just adding other softmax layers for other tasks. In our paper, only four datasets are used.

xuyige avatar Jun 02 '20 16:06 xuyige

For saving models, we did not save checkpoints during fine-tuning. If you need to save your models, we suggest using torch.save

xuyige avatar Jun 02 '20 16:06 xuyige

Ok, thanks for your reply!

swufelyc avatar Jun 03 '20 00:06 swufelyc