3 ms·
> Finetuning a pretrained model using your own dataset can be done much more quickly (think hours to a day or two) and on consumer-level hardware (I train my mo
by codetrotter 5y ago
> Finetuning a pretrained model using your own dataset can be done much more quickly (think hours to a day or two) and on consumer-level hardware (I train my models on an old desktop with a GTX 1070).
This is interesting! Do you have some links about doing that?
My desktop computer has a GTX 1060 with 6 GB of VRAM. But hopefully I can use it for something like this.
I've only used Google Colab in the past, and only tried stuff with prompting existing models.
Would love to experiment a bit with fine-tuning models on my own datasets to get some kind of unique stuff.
- godelski 5y agoIt is the same transfer learning you would do with any model. StyleGAN (and StyleGAN2-ADA) provide pretrained weights for you. Just start there and train on the new dataset. The ADA github even has tools to format your dataset correctly.