预测模型时空预测模型PyTorch复现
预测模型时空预测模型PyTorch复现models文件夹在models目录中每一个文件夹存储一个结构的完整模型代码复现参照了论文中的公式图示以及GitHub作者实现的代码如果有的话这些模型均假定输入的Tensor的shape为batch sequence channel height width这里的目的是为了学习尽可能内聚成一个个小的Module再组合的应该效率很差util文件夹patch针对大尺寸数据进行patch分割的方法不过这里要根据实际情况修改下这里是针对五维数据的如果针对四维则参照逻辑修改下即可TrainingTemplate和TestingTemplate我自己写的训练过程的模板类一般继承重写一些方法即可content tree包含生成目录树的方法
下载地址
用户评论
喜欢这种PyTorch复现的文章，比看别人的代码自己慢慢理解要来得快。
谢谢作者，让我有机会学到这种新的预测方式。
看了一遍，感觉我的PyTorch水平又上了一个台阶。
这份文件很详细地介绍了时空预测模型PyTorch的实现方法，值得一读。