Skip to content

Commit e4cc619

Browse files
authored
Update README.md
1 parent c44b6cb commit e4cc619

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,11 @@
33

44
We started from [this](https://github.com/rogertrullo/pytorch_convlstm/blob/master/conv_lstm.py) implementation and heavily refactored it add added features to match our needs.
55

6+
Please note that in this repository we implement the following dynamics:
7+
![CLSTM_dynamics](https://user-images.githubusercontent.com/7113894/59357391-15c73e00-8d2b-11e9-8234-9d51a90be5dc.png)
8+
9+
which is a bit different from the one in the original [paper](https://arxiv.org/pdf/1506.04214.pdf).
10+
611
### How to Use
712
The `ConvLSTM` module derives from `nn.Module` so it can be used as any other PyTorch module.
813

0 commit comments

Comments
 (0)