Thanks for your great work. The result is surprising. So, I try to train or finetune your model using danbooru2000 dataset. Unfortunately, I find that the loss hardly decrease no matter train from scratch or finetune your model. I train it with batch size 16 in 2 * V100 GPUs. And then, I add a controlnet, the loss is also hard to decrease. Can you tell me your loss trend of change and give me some advices? Below is my tensorboard. Thanks in advance.
Thanks for your great work. The result is surprising. So, I try to train or finetune your model using danbooru2000 dataset. Unfortunately, I find that the loss hardly decrease no matter train from scratch or finetune your model. I train it with batch size 16 in 2 * V100 GPUs. And then, I add a controlnet, the loss is also hard to decrease. Can you tell me your loss trend of change and give me some advices? Below is my tensorboard. Thanks in advance.