I am trying to develop a custom model. In my model, I have an LSTM where I am using the previous hidden and cell states
(h_0, c_0). So, if the number of samples isn’t divisible by the
batch_size, I am facing an error -
RuntimeError: Expected hidden size (1, 3, 256), got [1, 8, 256].
Torch Dataloader has an option
drop_last. How can I achieve such functionality with