Skip to content

Commit 9f662ba

Browse files
authored
Merge pull request #14 from DansYU/patch-1
Update README.md
2 parents 3607865 + e413c50 commit 9f662ba

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

README.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,7 @@ The ConvLSTM class supports an arbitrary number of layers. In this case, it can
1515

1616
Example usage:
1717
```
18-
model = ConvLSTM(input_size=(height, width),
19-
input_dim=channels,
18+
model = ConvLSTM(input_dim=channels,
2019
hidden_dim=[64, 64, 128],
2120
kernel_size=(3, 3),
2221
num_layers=3,

0 commit comments

Comments
 (0)