()
| 128 | |
| 129 | |
| 130 | def parse_args(): |
| 131 | parser = argparse.ArgumentParser() |
| 132 | parser.add_argument("meta_path", type=str, help="Path to the input CSV file") |
| 133 | parser.add_argument("--bs", type=int, default=1, help="Batch size") # don't use too large bs for unimatch |
| 134 | parser.add_argument("--num_workers", type=int, default=16, help="Number of workers") |
| 135 | parser.add_argument("--skip_if_existing", action="store_true") |
| 136 | args = parser.parse_args() |
| 137 | return args |
| 138 | |
| 139 | |
| 140 | @torch.no_grad() |