3 ms·
Question, how long did it take to train this model and what hardware did you use?
by xigency 3y ago
Question, how long did it take to train this model and what hardware did you use?
- erwannmillon 3y agoTook a lot of failed experiments, the model would keep converging to greyscale / sepia images. Think one of the ways I fixed was by adding an greyscale encoder to the arch. Used its output embedding as additional conditioning. Can't remember if I only added it to the Unet input or injected it during various stages of the unet down pass.
- erwannmillon 3y agoThink the final training run was only a couple hours on a Colab V100