| 25 | |
| 26 | |
| 27 | def parse_args(): |
| 28 | parser = argparse.ArgumentParser() |
| 29 | parser.add_argument("--config") |
| 30 | parser.add_argument("-W", type=int, default=512) |
| 31 | parser.add_argument("-H", type=int, default=784) |
| 32 | parser.add_argument("-L", type=int, default=24) |
| 33 | parser.add_argument("--seed", type=int, default=42) |
| 34 | parser.add_argument("--cfg", type=float, default=3.5) |
| 35 | parser.add_argument("--steps", type=int, default=30) |
| 36 | parser.add_argument("--fps", type=int) |
| 37 | args = parser.parse_args() |
| 38 | |
| 39 | return args |
| 40 | |
| 41 | |
| 42 | def main(): |