Skip to content

Commit 4f9b5f2

Browse files
author
zihao.chen_tp
committed
init
1 parent 85e71b8 commit 4f9b5f2

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,9 @@
11
# RNN_Pytorch
22
使用pytorch框架搭建一个encoder-forecaster结构的图像时序预测模型,ConvGRU内核
33

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

14-
模型说明
15+
##模型说明
1516

1617
本项目参考的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)
1718
encoder和forecaster 结构皆参照此论文编写

0 commit comments

Comments
 (0)