From 62a950ce34be100ac8f1d8199ae6a319bacfd900 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Tue, 28 Apr 2020 10:40:42 +0200 Subject: [PATCH] removed unused vars --- pts/model/lstnet/lstnet_network.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pts/model/lstnet/lstnet_network.py b/pts/model/lstnet/lstnet_network.py index d8acccf..8f5d160 100644 --- a/pts/model/lstnet/lstnet_network.py +++ b/pts/model/lstnet/lstnet_network.py @@ -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)