About loading data files

[cg-6x]

Hi, dear Jiu, thank you for sharing the code.

I use the given data generation code to generate the corresponding. h5 file for 291 images, but running main_ vdsr.py ，the following errors will be reported,
===> Loading datasets
Traceback (most recent call last):
File "main_vdsr.py", line 138, in
main()
File "main_vdsr.py", line 54, in main
training_data_loader = DataLoader(dataset=train_set, num_workers=opt.threads, batch_size=opt.batchSize, shuffle=True)
File "/root/anaconda3/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 213, in init
sampler = RandomSampler(dataset)
File "/root/anaconda3/lib/python3.7/site-packages/torch/utils/data/sampler.py", line 92, in init
if not isinstance(self.num_samples, int) or self.num_samples <= 0:
File "/root/anaconda3/lib/python3.7/site-packages/torch/utils/data/sampler.py", line 100, in num_samples
return len(self.data_source)
File "/home/thw/Desktop/Cg/VDSR/pytorch-vdsr-master/dataset.py", line 16, in len
return self.data.shape[0]
AttributeError: 'NoneType' object has no attribute 'shape'
I confirm that my training data is generated according to the code for data generation. What is the reason?