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

作者基于 Graphcast 架构，用不到 1000 行代码搭建了一个小型 AI 天气模型，无需超级计算机即可运行。模型采用编码-处理-解码结构，以 GNN 为核心，在 ERA5 数据集的经纬度网格上训练，内部用二十面体网格做潜在表示。相比需数百万行代码、单次预报耗时近一小时的 NWP 模型，这类数据驱动模型效率高出数个数量级。