BucketIterator in Pytorch is putting the batch_size on the torch.shape[1]
Hello guys I m trying to understand how to make a forward pass on an encoder decoder model with a custom dataset.
I have created a BucketIterator to see how the tensor.shape looks like for a batch of size 64.