Pre trained models - Githubissues

ProGamerGov / dream-creator

Quickly and easily create / train a custom DeepDream model

MIT License

64 stars 6 forks source link

Pre trained models #20

Open aertist opened 3 years ago

aertist commented 3 years ago

Can i have those models you trained? 👀

ProGamerGov commented 3 years ago

The summer model was accidentally trained with the wrong colorspace, and the others had a layer in the wrong place (both issues have been fixed since then). So I'm not sure if they're worth sharing?

aertist commented 3 years ago

How much time it took to train each of them? And Nvidia gpu you used for this purpose? 😅

ProGamerGov commented 3 years ago

I mostly used a 1050ti (Google Colab was used for earlier experiments until I realized that I could do it on my laptop) and it took a couple of hours at the most. Training time though really depends on the size of your dataset.