Earth-Forecasting-Transformer 项目使用教程

Earth-Forecasting-Transformer 项目使用教程

earth-forecasting-transformer Official implementation of Earthformer 项目地址: https://gitcode.com/gh_mirrors/ea/earth-forecasting-transformer

1. 项目目录结构及介绍

earth-forecasting-transformer/
├── datasets/
│   └── nbody/
├── figures/
├── scripts/
│   └── datasets/
│       ├── nbody/
│       └── sevir/
├── src/
│   └── earthformer/
├── tests/
├── .gitignore
├── CODE_OF_CONDUCT.md
├── CONTRIBUTING.md
├── LICENSE
├── NOTICE
├── README.md
└── setup.py

目录结构介绍

• datasets/: 存放数据集的目录，例如 nbody 数据集。
• figures/: 存放项目相关的图表和图片。
• scripts/: 包含数据集下载和处理的脚本，例如 nbody 和 sevir 数据集的下载脚本。
• src/earthformer/: 项目的核心代码，包含 Earthformer 模型的实现。
• tests/: 存放测试代码的目录。
• .gitignore: Git 忽略文件配置。
• CODE_OF_CONDUCT.md: 项目的行为准则。
• CONTRIBUTING.md: 贡献指南。
• LICENSE: 项目许可证。
• NOTICE: 项目通知文件。
• README.md: 项目介绍和使用说明。
• setup.py: 项目的安装配置文件。

2. 项目的启动文件介绍

项目的启动文件主要集中在 src/earthformer/ 目录下。以下是一些关键的启动文件：

• src/earthformer/main.py: 这是项目的入口文件，负责初始化模型、加载数据集并启动训练或推理过程。
• src/earthformer/train.py: 训练脚本，负责模型的训练过程。
• src/earthformer/inference.py: 推理脚本，负责模型的推理过程。

启动文件的使用

# 启动训练
python src/earthformer/train.py --config path/to/config.yaml

# 启动推理
python src/earthformer/inference.py --config path/to/config.yaml

3. 项目的配置文件介绍

项目的配置文件主要用于定义模型的超参数、数据集路径、训练参数等。配置文件通常使用 YAML 格式。

配置文件示例

# config.yaml
model:
  name: Earthformer
  layers: 12
  hidden_size: 768
  num_heads: 12

dataset:
  name: SEVIR
  path: datasets/sevir
  batch_size: 32

training:
  epochs: 100
  learning_rate: 0.0001
  optimizer: Adam

配置文件的使用

在启动训练或推理时，通过 --config 参数指定配置文件的路径：

python src/earthformer/train.py --config path/to/config.yaml

通过以上步骤，您可以顺利地启动和配置 Earth-Forecasting-Transformer 项目。

earth-forecasting-transformer Official implementation of Earthformer 项目地址: https://gitcode.com/gh_mirrors/ea/earth-forecasting-transformer