Skip to content

Commit e14aa13

Browse files
author
zihao.chen_tp
committed
init
1 parent 13ffaad commit e14aa13

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

README.md

+4-3
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
# RNN_Pytorch
22
使用pytorch框架搭建一个encoder-forecaster结构的图像时序预测模型,ConvGRU内核
33

4-
#文件说明
5-
BMSELoss.py 一个对不同降雨级别进行加权损失的loss函数。
4+
文件说明
5+
./BMSELoss.py 一个对不同降雨级别进行加权损失的loss函数。
66
ConvGRUCell.py 利用卷积操作实现的ConvGRU内核,单Cell
77
encoder。py 序列编码结构
88
forecaster.py 序列预测结构
@@ -11,7 +11,8 @@
1111
RNN.py 一般的RNN模型
1212
RNN_train.py RNN模型的训练
1313

14-
#模型说明
14+
模型说明
15+
1516
本项目参考的HKO模型为论文[Deep learning for precipitation nowcasting: A benchmark and a new model](http://papers.nips.cc/paper/7145-deep-learning-for-precipitation-nowcasting-a-benchmark-and-a-new-model)
1617
encoder和forecaster 结构皆参照此论文编写
1718
本文的数据样本原始大为20*1*477*477,训练过程中会缩放或随机切割到120*120大小。

0 commit comments

Comments
 (0)