diff --git a/metnet/layers/ConvGRU.py b/metnet/layers/ConvGRU.py index 1be2041..84432a7 100755 --- a/metnet/layers/ConvGRU.py +++ b/metnet/layers/ConvGRU.py @@ -189,7 +189,8 @@ def forward(self, input, hidden_state=None): input_tensor: 5-D Tensor either of shape (t, b, c, h, w) or (b, t, c, h, w) hidden_state: - Returns + + Returns: ------- last_state_list, layer_output """