New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

evaluation during training causes the stucking #12

Open

garychan22 opened this issue Jun 25, 2022 · 0 comments

garychan22 commented Jun 25, 2022 •

edited

Loading

Hi, thanks for your excellent project !

I want to evaluate the model during training by adding the following code in the train.py script, which leads to program stucking.

if i % conf.evaluate.valid_every == 0:
    model.eval()
    img = diffusion.p_sample_loop(model, [16, 3, 256, 256], model.device)
    save_image(img, "sample.png", normalize=True, range=(-1,1), nrow=4)
    model.train()

Appreciate any help !

The text was updated successfully, but these errors were encountered:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment