search

hendrycks / imagenet-r

ImageNet-R(endition) and DeepAugment (ICCV 2021)
MIT License
254 stars 18 forks source link

EDSR and CAE at once while training #6

Closed khawar-islam closed 2 years ago

khawar-islam commented 2 years ago

Hi @hendrycks

Can i choose choices=['edsr', 'cae'] both datasets at once while training? I am getting an error that you cannot choose two datasets

hendrycks commented 2 years ago

They can be put together. train_dataset = torch.utils.data.ConcatDataset([train_dataset, edsr_dataset, cae_dataset])