'Subset' object has no attribute 'train'

[ptrblck]

You could try to use this code to remap the class indices.