While model training on custom data i got this error
Traceback (most recent call last):
File "/notebooks/photo2cartoon/train.py", line 84, in <module>
main()
File "/notebooks/photo2cartoon/train.py", line 75, in main
gan.train()
File "/notebooks/photo2cartoon/models/UGATIT_sadalin_hourglass.py", line 178, in train
real_A, _ = trainA_iter.next()
File "/usr/local/lib/python3.9/dist-packages/torch/utils/data/dataloader.py", line 652, in __next__
data = self._next_data()
File "/usr/local/lib/python3.9/dist-packages/torch/utils/data/dataloader.py", line 692, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/usr/local/lib/python3.9/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.9/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/notebooks/photo2cartoon/dataset.py", line 66, in __getitem__
sample = self.loader(path)
File "/notebooks/photo2cartoon/dataset.py", line 99, in default_loader
return pil_loader(path)
File "/notebooks/photo2cartoon/dataset.py", line 95, in pil_loader
return img.convert('RGB')
File "/usr/local/lib/python3.9/dist-packages/PIL/Image.py", line 934, in convert
self.load()
File "/usr/local/lib/python3.9/dist-packages/PIL/ImageFile.py", line 276, in load
raise_oserror(err_code)
File "/usr/local/lib/python3.9/dist-packages/PIL/ImageFile.py", line 71, in raise_oserror
raise OSError(message + " when reading image file")
OSError: unrecognized data stream contents when reading image file
While model training on custom data i got this error