ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 2048])

[a-ceron]

[a-ceron]

Parece que se solucionó solo, no voy a cerrar el issue porqué aún estoy haciendo pruebas

[a-ceron]

Reduje la cantidad de imágenes de entrada para solo considerar mil, con esta configuración el error ya no aparece.

Puede que en los datos, no todos tienen el mismo formato/tamaño. Aunque en la función de transform tengo configurado un resize para solucionar esto. Además en el dataset se convierten en RGB para tener tres canales.

Update

Esta tomando el mismo comportamiento que con los archivos descargados.

• Podría ser problema del código.
• Podría ser la cantidad de archivos (muy pocos).