时空序列预测模型采用 PyTorch 框架实现。模型代码完整,复现论文公式和图示,支持输入 Tensor 形状:(batch, sequence, channel, height, width)。此外,还包含以下组件:

- Patch 分割方法,用于处理大尺寸数据

- 训练和测试模板类,可轻松自定义训练过程

- 文件夹生成函数,方便管理目录树。