Imagen-pytorch icon indicating copy to clipboard operation
Imagen-pytorch copied to clipboard

any plans for the training codes?

Open runzeer opened this issue 2 years ago • 5 comments

runzeer avatar Jul 15 '22 11:07 runzeer

+1

ghost avatar Aug 15 '22 18:08 ghost

I've had limited success modifying the code based off of this commit, but caveat emptor. I ended up deciding to wait for a future update, so don't have much additional context to add.

https://github.com/cene555/Imagen-pytorch/commit/d0e3e4be854ffcdd4ade6e18cac9e909e970ac81

lupinetine avatar Aug 15 '22 18:08 lupinetine

main purpose is to train from scratch, does not rely on interference lol.

ghost avatar Aug 15 '22 19:08 ghost

Training is possible under the commit referenced in my previous comment, if you're willing to modify the code to fit your specific environment. Inference code is included in the current (latest) commit.

lupinetine avatar Aug 19 '22 17:08 lupinetine

@lupinetine Did you get good training results?

FusionLi avatar Aug 23 '22 09:08 FusionLi