Hi @sharvil @Andrechang @JCBrouwer thanks for this implementation.
My issue is about the training time for unconditional generation. It takes me about 5 hours/ epoch on 1 * RTX8000 and most of the time is spent on loss.backward(), with the unconditional setting in #5, I wonder:
Is this common?
Any suggestions for acceleration please?
From how many epochs that you start to have good-quality generations?
Hi @sharvil @Andrechang @JCBrouwer thanks for this implementation.
My issue is about the training time for unconditional generation. It takes me about 5 hours/ epoch on 1 * RTX8000 and most of the time is spent on
loss.backward()
, with the unconditional setting in #5, I wonder:Thanks in advance.