Hi,
After some time in the training phase, I get the following error:
My batch_size = 32
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-22-3725f29f93d4> in <module>
     37 
     38         # get the output from the model
---> 39         output, h = mynet(specs, h)
     40         output = F.log_softmax(output, dim=2)
     41         output = output.transpose(0,1)

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py in __call__(self, *input, **kwargs)
    548             result = self._slow_forward(*input, **kwargs)
    549         else:
--> 550             result = self.forward(*input, **kwargs)
    551         for hook in self._forward_hooks.values():
    552             hook_result = hook(self, input, result)

<ipython-input-17-9a3df9ad77fe> in forward(self, x, hidden)
     45         x = x.transpose(1,2)
     46         #print("After View & Transpose : {}".format(x.size()))
---> 47         out, hidden = self.gru(x, hidden)
     48         #print("After GRU : {}".format(x.size()))
     49         #out = out.contiguous().view(-1, self.hidden_dim)

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/module.py in __call__(self, *input, **kwargs)
    548             result = self._slow_forward(*input, **kwargs)
    549         else:
--> 550             result = self.forward(*input, **kwargs)
    551         for hook in self._forward_hooks.values():
    552             hook_result = hook(self, input, result)

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/rnn.py in forward(self, input, hx)
    722             hx = self.permute_hidden(hx, sorted_indices)
    723 
--> 724         self.check_forward_args(input, hx, batch_sizes)
    725         if batch_sizes is None:
    726             result = _VF.gru(input, hx, self._flat_weights, self.bias, self.num_layers,

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/rnn.py in check_forward_args(self, input, hidden, batch_sizes)
    192         expected_hidden_size = self.get_expected_hidden_size(input, batch_sizes)
    193 
--> 194         self.check_hidden_size(hidden, expected_hidden_size)
    195 
    196     def permute_hidden(self, hx, permutation):

/opt/conda/lib/python3.7/site-packages/torch/nn/modules/rnn.py in check_hidden_size(self, hx, expected_hidden_size, msg)
    185         # type: (Tensor, Tuple[int, int, int], str) -> None
    186         if hx.size() != expected_hidden_size:
--> 187             raise RuntimeError(msg.format(expected_hidden_size, tuple(hx.size())))
    188 
    189     def check_forward_args(self, input, hidden, batch_sizes):

RuntimeError: Expected hidden size (5, 12, 128), got (5, 32, 128)

I guess this is because it is the last batch and isn’t 32 exactly. How can I avoid this?
Thanks.
BR,
Shweta.