diff --git a/DDPM/ddpm_conditional.py b/DDPM/ddpm_conditional.py index 3f343fd6da533dbfc7bd1e0b8a46ecbc5c0d6ab8..8f958e3de22c2c9a500dd2465b5bf417c4b93a75 100644 --- a/DDPM/ddpm_conditional.py +++ b/DDPM/ddpm_conditional.py @@ -178,7 +178,7 @@ def run(): parser = argparse.ArgumentParser() args = parser.parse_args() args.run_name = 'smoke' - args.epochs = 200 + args.epochs = 500 args.batch_size = 8 args.image_size = 64 args.num_classes = 10