error

[Mlegbder]

/root/miniconda3/envs/ISNet-cu118/lib/python3.8/site-packages/torch/nn/_reduction.py:42: UserWarning: size_average and reduce args will be deprecated, please use reduction='mean' instead. warnings.warn(warning.format(ret)) building model... batch size: 8 --- create training dataloader --- ------------------------------ train -------------------------------- --->>> train dataset 0 / 1 DIS5K-TR <<<--- -im- DIS5K-TR ../DIS5K/DIS-TR/im : 1 -gt- DIS5K-TR ../DIS5K/DIS-TR/gt : 1 Traceback (most recent call last): File "train_valid_inference_main.py", line 729, in main(train_datasets, File "train_valid_inference_main.py", line 535, in main train_dataloaders, train_datasets = create_dataloaders(train_nm_im_gt_list, File "/root/DIS-A100-4090/IS-Net/data_loader_cache.py", line 97, in create_dataloaders gos_dataloaders.append(DataLoader(gos_dataset, batch_size=batch_size, shuffle=shuffle, num_workers=numworkers)) File "/root/miniconda3/envs/ISNet-cu118/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 349, in init sampler = RandomSampler(dataset, generator=generator) # type: ignore[arg-type] File "/root/miniconda3/envs/ISNet-cu118/lib/python3.8/site-packages/torch/utils/data/sampler.py", line 140, in init raise ValueError(f"num_samples should be a positive integer value, but got num_samples={self.num_samples}") ValueError: num_samples should be a positive integer value, but got num_samples=0

[midone1]

Check to make sure you're connected to a GPU?