LAPE icon indicating copy to clipboard operation
LAPE copied to clipboard

How to fine-tune on my custom dataset

Open jmren168 opened this issue 2 years ago • 2 comments

Thanks for the awesome work.

I'm trying to fine-tune the MAST model on my custom dataset. Is it possible to do? Thanks.

jmren168 avatar Feb 16 '24 00:02 jmren168

Hi @jmren168 ,

Yes ofcourse! What commands have you tried and were unsuccessful? You can follow the upstream/downstream fine-tuning details in the readme and change encoder to MAST for fine-tuning!

Sreyan88 avatar Jul 09 '24 17:07 Sreyan88

I have added some details to the Readme for training MAST! Please let me know if it does not work!

Sreyan88 avatar Jul 09 '24 17:07 Sreyan88