Training

Overview

How to train a model for use with Bris-inference using anemoi-training

CPU training of a small test model

Running tox -e training should do this automatically. See Tests

• Fetch sample dataset from https://github.com/metno/bris-test-data/
• Create a config pointing to dataset based on tox_test_train.yaml.
• Install anemoi-training
• Run ANEMOI_BASE_SEED=1234 anemoi-training train --config-name tox_test_train.
• Your output will be a graph small_graph_20p0.pt and a checkpoint inference-last.ckpt for use in bris-inference.

Production training

Bris is trained on HPC's using metno/run-anemoi. See also anemoi-training and anemoi-datasets docs.