From 9ae33587fdf02a6b7ed0b29a1ffb94e368b38f62 Mon Sep 17 00:00:00 2001 From: Jesper Dramsch Date: Wed, 31 Jul 2024 17:21:26 +0200 Subject: [PATCH] copy config/debug.yaml Co-authored-by: Ana Prieto Nemesio <91897203+anaprietonem@users.noreply.github.com> Co-authored-by: Jesper Dramsch Co-authored-by: Mario Santa Cruz <48736305+JPXKQX@users.noreply.github.com> Co-authored-by: Matthew Chantry Co-authored-by: Simon Lang Co-authored-by: Sara Hahner --- src/anemoi/training/config/debug.yaml | 32 +++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 src/anemoi/training/config/debug.yaml diff --git a/src/anemoi/training/config/debug.yaml b/src/anemoi/training/config/debug.yaml new file mode 100644 index 00000000..5c12e23a --- /dev/null +++ b/src/anemoi/training/config/debug.yaml @@ -0,0 +1,32 @@ +defaults: +- data: zarr +- dataloader: native_grid +- diagnostics: eval_rollout +- hardware: example +- graph: default +- model: gnn +- training: default +- _self_ + + +### This file is for local experimentation. +## When you commit your changes, assign the new features and keywords +## to the correct defaults. +# For example to change from default GPU count: +# hardware: +# num_gpus_per_node: 1 + +diagnostics: + plot: + enabled: False +hardware: + files: + graph: ... +model: + num_channels: 128 +dataloader: + limit_batches: + training: 100 + validation: 100 +training: + max_epochs: 5