Mikubill / naifu

Train generative models with pytorch lightning
MIT License
284 stars 36 forks source link

Does it work? Output images are all noise. #1

Closed IdiotSandwichTheThird closed 1 year ago

IdiotSandwichTheThird commented 1 year ago

Training with default setting on colab, after 3 epochs the output images are pure noise, (see attachment). Earlier test at 90 epoch, same result. tmpnvb4bjko

Mikubill commented 1 year ago

Seems WebUI is not happy with the format of the checkpoint. Added a small conversion script and it should work now.

IdiotSandwichTheThird commented 1 year ago

It does work now, after running output.ckpt through convert.py.