| 135 | |
| 136 | |
| 137 | def get_args(default_batch=128, default_z_dim=100): |
| 138 | parser = argparse.ArgumentParser() |
| 139 | parser.add_argument('--gpu', help='comma separated list of GPU(s) to use.') |
| 140 | parser.add_argument('--load', help='load model') |
| 141 | parser.add_argument('--sample', action='store_true', help='view generated examples') |
| 142 | parser.add_argument('--data', help='a jpeg directory') |
| 143 | parser.add_argument('--load-size', help='size to load the original images', type=int) |
| 144 | parser.add_argument('--crop-size', help='crop the original images', type=int) |
| 145 | parser.add_argument( |
| 146 | '--final-size', default=64, type=int, |
| 147 | help='resize to this shape as inputs to network') |
| 148 | parser.add_argument('--z-dim', help='hidden dimension', type=int, default=default_z_dim) |
| 149 | parser.add_argument('--batch', help='batch size', type=int, default=default_batch) |
| 150 | global args |
| 151 | args = parser.parse_args() |
| 152 | if args.gpu: |
| 153 | os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu |
| 154 | return args |
| 155 | |
| 156 | |
| 157 | if __name__ == '__main__': |