🐛 Describe the bug

Setting ["bedroom"] to classes argument of LSUN() still gets the error message saying Valid values are {'bedroom', ...} as shown below:

from torchvision import datasets

train_data = datasets.LSUN(
    root="data/lsun",
    classes=["bedroom"] # Here
) # Error

ValueError: Unknown value '' for LSUN class. Valid values are {'bedroom', 'bridge', 'church_outdoor', 'classroom', 'conference_room', 'dining_room', 'kitchen', 'living_room', 'restaurant', 'tower'}.

But setting ["bedroom_train"] or ["bedroom_val"] to classes argument of LSUN() doesn't get the error as shown below:

from torchvision import datasets

train_data1 = datasets.LSUN(
    root="data/lsun",
    classes=["bedroom_train"] # Here
)

train_data2 = datasets.LSUN(
    root="data/lsun",
    classes=["bedroom_val"] # Here
)

And setting "train", "val" or "test" to classes argument of LSUN() doesn't get the error as shown below:

from torchvision import datasets

train_data1 = datasets.LSUN(
    root="data/lsun",
    classes="train" # Here
)

train_data2 = datasets.LSUN(
    root="data/lsun",
    classes"val" # Here
)

train_data3 = datasets.LSUN(
    root="data/lsun",
    classes"test" # Here
)

So, the error message should be like as shown below:

ValueError: Unknown value '' for LSUN class. The valid value is one of 'train', 'val' or 'test' or a list of categories e.g. ['bedroom_train', 'bedroom_val', 'bridge_train', 'bridge_val', 'church_outdoor_train', 'church_outdoor_val', 'classroom_train', 'classroom_val', 'conference_room_train', 'conference_room_val', 'dining_room_train', 'dining_room_val', 'kitchen_train', 'kitchen_val', 'living_room_train', 'living_room_val', 'restaurant_train', 'restaurant_val', 'tower_train', 'tower_val'].

Versions

import torchvision

torchvision.__version__ # '0.20.1'