Command-line interface

The torchgeo-bench console script exposes two subcommands:

torchgeo-bench run [hydra overrides...]

Runs the benchmark pipeline. All extra arguments are forwarded to Hydra, so any value in src/torchgeo_bench/conf/config.yaml (or any model preset under conf/model/) can be overridden directly on the command line, e.g. model=timm/resnet50 dataset.names=[m-eurosat].

torchgeo-bench download {geobench_v1|geobench_v2|eurosat}

Downloads benchmark datasets into ./data/ (or a custom location with --output-dir). For GeoBench V2, individual datasets can be selected with --datasets a,b,c.

Hydra entry point

The actual benchmark loop lives in torchgeo_bench.main and is decorated with @hydra.main:

torchgeo_bench.main.main(cfg)

Run the benchmark pipeline for all configured datasets and models.

Download helpers

torchgeo_bench.download.download_geobench_v1(output_dir)

Download GeoBench V1 to output_dir (creates classification_v1.0/ inside).

torchgeo_bench.download.download_geobench_v2(output_dir, datasets=None)

Download GeoBench V2 datasets into output_dir/geobenchv2/<name>.

Parameters:

• output_dir (Path) – Benchmark data root (typically data/).

• datasets (list[str] | None) – Specific dataset names to fetch. None downloads DEFAULT_V2_DATASETS.

torchgeo_bench.download.download_eurosat(output_dir)

Download torchgeo’s EuroSAT into output_dir/eurosat for all splits.