()
| 92 | |
| 93 | |
| 94 | def launch(): |
| 95 | import argparse |
| 96 | parser = argparse.ArgumentParser() |
| 97 | args = parser.parse_args() |
| 98 | args.run_name = "DDPM_Uncondtional" |
| 99 | args.epochs = 500 |
| 100 | args.batch_size = 12 |
| 101 | args.image_size = 64 |
| 102 | args.dataset_path = r"C:\Users\dome\datasets\landscape_img_folder" |
| 103 | args.device = "cuda" |
| 104 | args.lr = 3e-4 |
| 105 | train(args) |
| 106 | |
| 107 | |
| 108 | if __name__ == '__main__': |