Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AIFS graph recipe and model config #41

Open
pimmeerdink opened this issue Dec 19, 2024 · 0 comments
Open

AIFS graph recipe and model config #41

pimmeerdink opened this issue Dec 19, 2024 · 0 comments
Labels
enhancement New feature or request graphs

Comments

@pimmeerdink
Copy link

Is your feature request related to a problem? Please describe.

I'm trying to replicate results from the AIFS paper, but would need to understand the recipe of the graph in order to start training the model myself and working with it. Are there plans to release the graph recipe and model config for the AIFS model?

Describe the solution you'd like

Ideally, there would be a subdirectory with all the necessary input files to construct AIFS. Even better: a notebook or example code that defines the AIFS model from scratch.

Describe alternatives you've considered

I have looked into loading the model from the weights present in the huggingface checkpoint, however here we can load the resulting graph (._graph_data), but it is not clear what reproducible recipe led to this graph. The same goes for the model config.

Additional context

No response

Organisation

No response

@pimmeerdink pimmeerdink added the enhancement New feature or request label Dec 19, 2024
@JesperDramsch JesperDramsch transferred this issue from ecmwf/anemoi-graphs Dec 19, 2024
theissenhelen added a commit that referenced this issue Jan 6, 2025
* fix: change pre-cmmit autoupdate schedule to monthly

* fix: change the merge strategy for Changelog to Union

* fix: add .envrc to .gitignore

* ci: ignore pre-commit-config and readthedocs for changelog updates

* ci: fix to correct hpc workflow call

* fix: update precommit config

* chore: update pre-commits

* feat: add codeowners file

* chore: update dependencies

* ci: add hpc-config

* docs: changelog

* fix: respond to review comments

---------

Co-authored-by: Jesper Dramsch <[email protected]>
theissenhelen added a commit that referenced this issue Jan 6, 2025
* fix: change pre-cmmit autoupdate schedule to monthly

* fix: change the merge strategy for Changelog to Union

* fix: add .envrc to .gitignore

* ci: ignore pre-commit-config and readthedocs for changelog updates

* ci: fix to correct hpc workflow call

* fix: update precommit config

* chore: update pre-commits

* feat: add codeowners file

* chore: update dependencies

* ci: add hpc-config

* docs: changelog

* fix: respond to review comments

---------

Co-authored-by: Jesper Dramsch <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request graphs
Projects
None yet
Development

No branches or pull requests

2 participants