Unsupervised-Compositional-Concepts-Discovery icon indicating copy to clipboard operation
Unsupervised-Compositional-Concepts-Discovery copied to clipboard

Published 20 hours ago • verified publisher icon nanlliu

Training Time

Open veroveroxie opened this issue 1 year ago • 0 comments

Hi, Thank you so much for your great work! I really like your work and am trying to run it. I have questions about the code.

  1. Do you mind sharing your training time and computing resources for the experiments? I am currently using a single L40 and train my dataset with 100 images. The progress bar shows that it takes about 5 hours to finish. Is it close to your training time?

  2. I found that you optimize the parameters after each backward instead of under accelerator.sync_gradients. Is it intended or a potential bug? In Line 877-880.

accelerator.backward(loss) optimizer.step() lr_scheduler.step() optimizer.zero_grad()

  1. After training, how do we generate compositional images, like "a photo of "t1" and "t2""?

Thanks a lot!

veroveroxie avatar May 29 '24 01:05 veroveroxie