如何用不到 1000 行代码构建自己的小型 AI 天气模型

作者基于 Graphcast 架构，用不到 1000 行代码搭建了一个小型 GNN 天气模型，无需超级计算机即可运行。模型使用 ERA5 数据集，采用编码-处理-解码结构与二十面体网格，网格节点比例约为 25 个网格单元对应 1 个网格节点。