scripts/run_batch_generation.py

from dataset_generation.gridnet_hd_batch_generation import batch_generation_main

# Example:
# python scripts/run_batch_generation.py --config configs/config_gridnet_hd_dataset_generation.yaml
batch_generation_main()