本期内容分享一套全流程搭建训练气象深度学习模型的代码,并考虑在基础的预报预测领域的应用。
本期内容的目标是:尽量让看过本文的读者,可以直接在本地一键跑通如SmaAtUNet、UNet等卷积网络,ConvLSTM、TrajGRU、PhyDNet、STPhy等递归卷积网络,以及SegFormer等基于Transformer的网络。
本期part1以预报预测为例,后续part2将以降尺度为例。
总体来说:
1、从https://github.com/Tsingzao/MetNow下载代码至本地;
2、执行run.py或者run_recurrent.py即可开始训练模型。不同模型大致效果如下,