¿Cómo cambio mi propio conjunto de datos al formato de conjunto de datos MNIST?

data_transform_test = transforms.Compose([transforms.ToTensor(),
transforms.Grayscale(num_output_channels=1),
transforms.Resize((28, 28)),
transforms.Normalize((0.5,), (0.5,))
])

Esto es lo que intenté para mi implementación de un modelo basado en el conjunto de datos MNIST. No puedo cambiar la arquitectura completa del modelo porque eso es demasiado trabajo para mí. Pero mi código me está dando este error. ¿Ustedes tienen alguna sugerencia o se necesitan modificaciones adicionales?

“`

IndexError Traceback (most recent call last)
<ipython-input-59-ce644b7b7998> in <module>()
1 # training for 3 epochs
2 nepochs = 3
—-> 3 losses = train(capsulenet, criterion, optimizer, nepochs=nepochs)

<ipython-input-58-54eb5db28cd7> in train(capsulenet, criterion, optimizer, nepochs, printevery)
26
27 # reshape and get target class
—> 28 target = torch.eye(10).indexselect(dim=0, index=target)
29
30 if TRAINONGPU:

IndexError: el índice está fuera de rango en self.“`