czzhangheng
/
DCRNN
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed docstring

This commit is contained in:
Chintan Shah 2019-10-07 10:48:48 -04:00
parent 5509e9aae5
commit a5a1063160
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model/pytorch/dcrnn_model.py
View File

@ -153,7 +153,7 @@ class DCRNNModel(nn.Module, Seq2SeqAttrs):
seq2seq forward pass
:param inputs: shape (seq_len, batch_size, num_sensor * input_dim)
:param labels: shape (horizon, batch_size, num_sensor * output)
:param batches_seen: batches seen till date
:param batches_seen: batches seen till now
:return: output: (self.horizon, batch_size, self.num_nodes * self.output_dim)
"""
encoder_hidden_state = self.encoder(inputs)