When I reduce the number of samples, the following problems will occur

[peiyunju]

Traceback (most recent call last): File "hed.py", line 309, in main() File "hed.py", line 189, in main train(train_loader, net, opt, lr_schd, epoch, save_dir=join(output_dir, 'epoch-{}-train'.format(epoch))) File "hed.py", line 214, in train for batch_index, (images, edges) in enumerate(tqdm(train_loader)): File "C:\Users\Administrator\anaconda3\envs\37\lib\site-packages\tqdm\std.py", line 1195, in iter for obj in iterable: File "C:\Users\Administrator\anaconda3\envs\37\lib\site-packages\torch\utils\data\dataloader.py", line 521, in next data = self._next_data() File "C:\Users\Administrator\anaconda3\envs\37\lib\site-packages\torch\utils\data\dataloader.py", line 561, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "C:\Users\Administrator\anaconda3\envs\37\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "C:\Users\Administrator\anaconda3\envs\37\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in data = [self.dataset[idx] for idx in possibly_batched_index] File "F:\hed\datasets.py", line 43, in getitem edge = edge[np.newaxis, :, :] # Add one channel at first (CHW). TypeError: 'NoneType' object is not subscriptable