时空预测模型PyTorch复现models文件夹在models目录中每一个文件夹存储一个结构的完整模型代码复现参照了论文中的公式图示以及GitHub作者实现的代码如果有的话这些模型均假定输入的Tensor的shape为batch sequence channel height width这里的目的是为了学习尽可能内聚成一个个小的Module再组合的应该效率很差util文件夹patch针对大尺寸数据进行patch分割的方法不过这里要根据实际情况修改下这里是针对五维数据的如果针对四维则参照逻辑修改下即可TrainingTemplate和TestingTemplate我自己写的训练过程的模板类一般继承重写一些方法即可content tree包含生成目录树的方法