BERT4doc-Classification
BERT4doc-Classification copied to clipboard
How to fine tuning model on multi-tasks?
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!
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.
For saving models, we did not save checkpoints during fine-tuning. If you need to save your models, we suggest using torch.save
Ok, thanks for your reply!