()
| 110 | |
| 111 | |
| 112 | def launch(): |
| 113 | import argparse |
| 114 | parser = argparse.ArgumentParser() |
| 115 | args = parser.parse_args() |
| 116 | args.run_name = "DDPM_conditional" |
| 117 | args.epochs = 300 |
| 118 | args.batch_size = 14 |
| 119 | args.image_size = 64 |
| 120 | args.num_classes = 10 |
| 121 | args.dataset_path = r"C:\Users\dome\datasets\cifar10\cifar10-64\train" |
| 122 | args.device = "cuda" |
| 123 | args.lr = 3e-4 |
| 124 | train(args) |
| 125 | |
| 126 | |
| 127 | if __name__ == '__main__': |
no test coverage detected