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

Commit

Permalink
Update transformer.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
gabrieloks authored Oct 28, 2024
1 parent b871132 commit a29bf91
Showing 1 changed file with 14 additions and 6 deletions.
20 changes: 14 additions & 6 deletions src/anemoi/training/config/model/transformer.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -53,12 +53,20 @@ node_loss_weight: area_weight

# Bounding configuration
bounding: #These are applied in order

# Bound tp (total precipitation) with a Relu bounding layer
# ensuring a range of [0, infinity) to avoid negative precipitation values.
- _target_: anemoi.models.layers.bounding.ReluBounding #[0, infinity)
variables:
- tp
- _target_: anemoi.models.layers.bounding.FractionBounding # fraction of tp
variables:
- cp
min_val: 0
max_val: 1
total_var: tp

# [OPTIONAL] Bound cp (convective precipitation) as a fraction of tp.
# This guarantees that cp is physically consistent with tp by restricting cp
# to a fraction of tp [0 to 1]. Uncomment the lines below to apply.

# - _target_: anemoi.models.layers.bounding.FractionBounding # fraction of tp
# variables:
# - cp
# min_val: 0
# max_val: 1
# total_var: tp

0 comments on commit a29bf91

Please sign in to comment.