Tengo que seleccionar un lote de 10 imágenes del conjunto de datos mnist. Cada imagen deberá pertenecer a una clase diferente, es decir, la imagen 0 a la clase 0, la imagen 1 a la clase 1, etc. Sé que de la siguiente manera extraigo todo el conjunto de datos. . . . Read more
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. . . . Read more