La clave del problema es cómo determinar la correspondencia entre imágenes y categorías.
from torchvision.datasets import ImageFolder
dataset = ImageFolder('F:/my_code/data/0dB/train/')
Leer en datos
print(dataset.imgs[2])
En este punto, dataset.imgs es una lista con muchos elementos, y el segundo
está seleccionado y la salida es
('F:/my_code/data/0dB/train/BFSK\\BFSK100.jpg', 0)
Es decir, la carpeta de imágenes debe organizarse en el orden del nombre de la carpeta,
BFSK corresponde a 0,
BPSK corresponde a 1,
etc ...