v-diffusion-pytorch icon indicating copy to clipboard operation
v-diffusion-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon crowsonkb

Training

Open demegire opened this issue 3 years ago • 4 comments

Is there a way to train custom diffusion models?

demegire avatar Jan 06 '22 12:01 demegire

It would be good to at least get a reply what the reason is for not publishing the training code. If it is not intended to be shared - fair enough. But it would be great to know....

Meanwhile did anyone figure out how to train?

xavierleung avatar Feb 10 '22 12:02 xavierleung

No I just wanted to clean it up first before releasing it and then never got around to it... ^^;;

crowsonkb avatar Feb 10 '22 20:02 crowsonkb

It is similar to my CIFAR-10 training notebook (https://colab.research.google.com/drive/1rKa8P8Sg1C8q2fzyiM4WMWpZ6sq86423) except PyTorch Lightning/distributed data parallel.

crowsonkb avatar Feb 10 '22 20:02 crowsonkb

No I just wanted to clean it up first before releasing it and then never got around to it... ^^;;

We would appreciate the dirty code too! It is better than nothing after all. In fact I could try to clean the code myself and merge it here if you published it. Thank you for your great contributions to the AI-Art community btw

demegire avatar Mar 09 '22 10:03 demegire