pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard

Finetuning the pretrained models

Open 17Skye17 opened this issue 6 years ago • 3 comments

Hi Cadene, nice work! 👍
Is there any efficient way to fine-tune these models on other datasets?

17Skye17 avatar Aug 22 '19 11:08 17Skye17

It's actually not too hard to write some boilerplate code to get a fine-tuning setup up and running or if you need something more solid you can use one of the many training frameworks that already exist on top of pytorch.

achaiah avatar Aug 23 '19 03:08 achaiah

@achaiah Thank you for comment. I built a new training framework based on this repo :-)

17Skye17 avatar Aug 24 '19 07:08 17Skye17

@achaiah Thank you for comment. I built a new training framework based on this repo :-)

巨佬的finetuning框架会开源在Git上嘛?

Magsun avatar Sep 25 '19 08:09 Magsun