基于PyTorch的时空预测模型复现与模型结构
在models文件夹中,存储了复现的时空预测模型PyTorch代码,每个文件夹包含一个完整的模型结构,复现依据论文公式、图示,以及GitHub作者的实现代码(如果有)。这些模型假定输入Tensor的shape为(batch, sequence, channel, height, width),旨在学习,通过内聚小模块再组合以实现效率。
下载地址
用户评论