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