Skip to content
This repository has been archived by the owner on Dec 20, 2024. It is now read-only.

Commit

Permalink
Add details to config
Browse files Browse the repository at this point in the history
  • Loading branch information
HCookie committed Dec 16, 2024
1 parent f51fbb1 commit 93d0e74
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion src/anemoi/training/config/training/default.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -58,19 +58,25 @@ loss_gradient_scaling: False

# Validation metrics calculation,
# This may be a list, in which case all metrics will be calculated
# and logged according to their name
# and logged according to their name.
# These metrics are calculated in the output model space, and thus
# have undergone postprocessing.
validation_metrics:
# loss class to initialise
- _target_: anemoi.training.losses.mse.WeightedMSELoss
# Scalars to include in loss calculation
# Cannot scale over the variable dimension due to possible remappings.
# Available scalars include:
# - 'loss_weights_mask': Giving imputed NaNs a zero weight in the loss function
# Use the `scale_validation_metrics` section to variable scale.
scalars: []
# other kwargs
ignore_nans: True

# List of validation metrics to keep in normalised space, and scalars to be applied
# Use '*' in reference all metrics, or a list of metric names.
# Unlike above, variable scaling is possible due to these metrics being
# calculated in the same way as the training loss, within the internal model space.
scale_validation_metrics:
scalars_to_apply: ['variable']
metrics:
Expand Down

0 comments on commit 93d0e74

Please sign in to comment.