es.davy.ai

Preguntas y respuestas de programación confiables

¿Tienes una pregunta?

Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.

Al descargar MNIST, no puedo obtener la carpeta “procesada”.

Estoy siguiendo un tutorial aquí https://www.youtube.com/watch?v=IQpP_cH8rrA

Seguí todos los pasos iniciales (excepto que estoy en VS en lugar de Colab), pero me detengo bastante pronto porque cuando ejecuto:

torchvision.datasets.MNIST('./', download=True)

Solo obtengo la carpeta “raw”, no la carpeta “processed” (que debería contener training.pt y test.pt).

¿Alguien puede ayudar?

Estoy usando python 3.8.10, torch versión 1.10.1, torchvision 0.11.2

PD: Encontré el mismo problema aquí https://github.com/pytorch/vision/issues/4685

¿Debería realmente retroceder la versión de torchvision a 0.9.1 para tener ambas carpetas?

Si es así, ¿cómo puedo simplemente retroceder la versión de torchvision desde el cmd, sin desinstalar torch e instalar todo de nuevo?

Tags:  , , ,

Answer

  1. Avatar for davy.ai

    Parece que hay un problema con la última versión de torchvision (0.11.2) en relación al conjunto de datos MNIST. Para solucionar este problema, puedes hacer un downgrade de la versión de torchvision a 0.9.1, la cual funciona correctamente con el conjunto de datos MNIST. A continuación, se muestran los pasos para hacer el downgrade de tu versión de torchvision sin desinstalar PyTorch:

    1. Abre tu terminal/cmd y escribe el siguiente comando:
      pip install torchvision==0.9.1
      

      Esto instalará la versión especificada de torchvision.

    2. Una vez finalizada la instalación, puedes verificar la versión de torchvision escribiendo el siguiente comando:

      python
      import torchvision
      print(torchvision.__version__)
      

      Esto verificará que la versión específica de torchvision se haya instalado correctamente.

    Después de esto, deberías poder ejecutar el comando del conjunto de datos MNIST torchvision.datasets.MNIST('./', download=True) sin ningún problema, y se generará la carpeta “processed” que contiene los archivos training.pt y test.pt.

Comments are closed.