Add Climateformer for climate prediction #1211
Open
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
使用pre-commit对代码格式化。
add Climateformer model for climate prediction add docs for Climateformer add examples for Climateformer add era5climate_dataset for Climateformer
|
Thanks for your contribution! |
HydrogenSulfate
requested changes
Sep 1, 2025
| USE_SAMPLED_DATA: false | ||
|
|
||
| # set train data path | ||
| TRAIN_FILE_PATH: /data/jinqizhao/ERA5/ |
| DATA_STD_PATH: ./datasets/era5/stat/std.nc | ||
|
|
||
| # set evaluate data path | ||
| VALID_FILE_PATH: /data/jinqizhao/ERA5/ |
examples/climateformer/main.py
Outdated
Comment on lines
80
to
82
| "sampler": { | ||
| "name": "BatchSampler", | ||
| }, |
There was a problem hiding this comment.
可删除
Suggested change
| "sampler": { | |
| "name": "BatchSampler", | |
| }, |
examples/climateformer/main.py
Outdated
Comment on lines
139
to
141
| "sampler": { | ||
| "name": "BatchSampler", | ||
| }, |
There was a problem hiding this comment.
可删除
Suggested change
| "sampler": { | |
| "name": "BatchSampler", | |
| }, |
There was a problem hiding this comment.
跟develop合并一下，不要删除无关代码
ppsci/arch/climateformer.py
Outdated
| # MultiDecoder | ||
| class Decoder(nn.Layer): | ||
| def __init__(self, C_hid: int, C_out: int, N_S: int): | ||
| super(Decoder, self).__init__() |
There was a problem hiding this comment.
Suggested change
| super(Decoder, self).__init__() | |
| super().__init__() |
There was a problem hiding this comment.
请在mkdocs.yml里添加当前案例文档
There was a problem hiding this comment.
请在arch.md中加入Climateformer
There was a problem hiding this comment.
请在dataset.md中加入ERA5ClimateDataset
Changes for Climateformer
liaoxin2
reviewed
Sep 22, 2025
docs/zh/examples/climateformer.md
Outdated
| @@ -0,0 +1,265 @@ | |||
| # Climateformer | |||
|
|
|||
| 开始训练、评估前，请下载ERA5数据集文件 | |||
docs/zh/examples/climateformer.md
Outdated
| === "模型评估命令" | ||
|
|
||
| ``` sh | ||
| python main.py mode=eval EVAL.pretrained_model_path=./outputs_climateformer/checkpoints/best_model.pdparams |
There was a problem hiding this comment.
模型已经上传，将./outputs_climateformer/checkpoints/best_model.pdparams修改为https://paddle-org.bj.bcebos.com/paddlescience/models/climateformer/climateformer.pdparams
docs/zh/examples/climateformer.md
Outdated
|
|
||
| 开始训练、评估前，请下载ERA5数据集文件 | ||
|
|
||
| 开始评估前，请下载或训练生成预训练模型 |
docs/zh/examples/climateformer.md
Outdated
| --8<-- | ||
| examples/climateformer/conf/climateformer.yaml | ||
| --8<-- | ||
| ``` No newline at end of file |
Revised Version of Climateformer for climate prediction
HydrogenSulfate
previously approved these changes
Oct 19, 2025
docs/zh/examples/climateformer.md
Outdated
| --8<-- | ||
| ``` | ||
|
|
||
| #### 3.2.6 模型导出 |
examples/climateformer/main.py
Outdated
| epochs=cfg.TRAIN.epochs, | ||
| iters_per_epoch=ITERS_PER_EPOCH, | ||
| log_freq=cfg.log_freq, | ||
| eval_during_train=True, |
There was a problem hiding this comment.
这里建议改成cfg.TRAIN.eval_during_train
liaoxin2
reviewed
Oct 19, 2025
docs/index.md
Outdated
| |-----|---------|-----|---------|----|---------|---------| | ||
| | 化学分子生成 | [Moflow](./zh/examples/moflow.md) | 数据驱动 | moflow | 监督学习 | qm9/ zink250k | [MoFlow: An Invertible Flow Model for Generating Molecular Graphs](https://arxiv.org/abs/2006.10137v1) | | ||
| | 化学反应预测 | [IFM](./zh/examples/ifm.md) | 数据驱动 | IFM-MLP | 监督学习 | tox21/sider/hiv/bace/bbbp | [Understanding the Limitations of Deep Models for Molecular property prediction: Insights and Solutions](https://openreview.net/pdf?id=NLFqlDeuzt) | | ||
| | 天气预报 | [Climateformer 气候预测](./zh/examples/climateformer.md) | 数据驱动 | Transformer | 监督学习 | [ERA5](https://https://cds.climate.copernicus.eu/datasets/reanalysis-era5-pressure-levels?tab=download) | - | |
均已pre-commit过了。
|
@zhangpu00000 yaml文件也需要用pre-commit过一遍 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Others
PR changes
Others
Describe
add Climateformer model for climate prediction
add docs for Climateformer
add examples for Climateformer
add era5climate_dataset for Climateformer