Skip to content

Add conditional layer norm to fe#2155

Draft
ankitpatnala wants to merge 3 commits intoecmwf:developfrom
ankitpatnala:ankit_add_conditional_layer_norm_to_fe
Draft

Add conditional layer norm to fe#2155
ankitpatnala wants to merge 3 commits intoecmwf:developfrom
ankitpatnala:ankit_add_conditional_layer_norm_to_fe

Conversation

@ankitpatnala
Copy link
Copy Markdown
Contributor

Description

Closes #2153

Added new config to the forecasting_engine.

It passes aux_info to the forecasting engine ( time_window_handler.start.day, time_window_handler.start.time, time_window_handler.end.day,time_window_handler.end.time )

Is this PR a draft? Mark it as draft.

Checklist before asking for review

  • I have performed a self-review of my code
  • My changes comply with basic sanity checks:
    • I have fixed formatting issues with ./scripts/actions.sh lint
    • I have run unit tests with ./scripts/actions.sh unit-test
    • I have documented my code and I have updated the docstrings.
    • I have added unit tests, if relevant
  • I have tried my changes with data and code:
    • I have run the integration tests with ./scripts/actions.sh integration-test
    • (bigger changes) I have run a full training and I have written in the comment the run_id(s): launch-slurm.py --time 60
    • (bigger changes and experiments) I have shared a hegdedoc in the github issue with all the configurations and runs for this experiments
  • I have informed and aligned with people impacted by my change:
    • for config changes: the MatterMost channels and/or a design doc
    • for changes of dependencies: the MatterMost software development channel

@ankitpatnala
Copy link
Copy Markdown
Contributor Author

cc @SavvasMel.
It is not so clean implementation. It is just for testing and there are some magic numbers like (aux_dim)

@ankitpatnala ankitpatnala changed the title Ankit add conditional layer norm to fe Add conditional layer norm to fe Mar 31, 2026
@github-actions github-actions bot added data Anything related to the datasets used in the project model Related to model training or definition (not generic infra) labels Mar 31, 2026
@ankitpatnala ankitpatnala mentioned this pull request Apr 16, 2026
4 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

data Anything related to the datasets used in the project model Related to model training or definition (not generic infra)

Projects

Status: No status

Development

Successfully merging this pull request may close these issues.

Task: Add time condtioning to the forecasting engine

2 participants