I’m trying to create a custom dataset in PyTorch that is similar to MNIST but with my own images. However, I’m having difficulties in preparing the dataset and possibly in the architecture of the seq2seq(encoder-decoder).