wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-27 11:26:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

removed unused vars

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-04-28 10:40:42 +02:00
parent 7e6ed1d87a
commit 62a950ce34
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/model/lstnet/lstnet_network.py
+2 -3
View File
@@ -65,13 +65,12 @@ class LSTNetBase(nn.Module):
"LSTM",
], "`skip_rnn_cell_type` must be either 'GRU' or 'LSTM' "
self.conv_out = context_length - kernel_size
self.conv_skip = self.conv_out // skip_size
conv_out = context_length - kernel_size
self.conv_skip = conv_out // skip_size
assert self.conv_skip > 0, (
"conv1d output size must be greater than or equal to `skip_size`\n"
"Choose a smaller `kernel_size` or bigger `context_length`"
)
self.skip_rnn_c_dim = channels * skip_size
self.cnn = nn.Conv2d(
in_channels=1, out_channels=channels, kernel_size=(num_series, kernel_size)