runtime error

Exit code: 1. Reason: 00:00, 1.03 examples/s] Traceback (most recent call last): File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 171, in collate { File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 172, in <dictcomp> key: collate( File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 240, in collate raise TypeError(default_collate_err_msg_format.format(elem_type)) TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.PngImagePlugin.PngImageFile'> During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/app.py", line 49, in <module> for batch in dataloader: File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 701, in __next__ data = self._next_data() File "/usr/local/lib/python3.10/site-packages/torch/utils/data/dataloader.py", line 757, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/fetch.py", line 55, in fetch return self.collate_fn(data) File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 398, in default_collate return collate(batch, collate_fn_map=default_collate_fn_map) File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 191, in collate return { File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 192, in <dictcomp> key: collate([d[key] for d in batch], collate_fn_map=collate_fn_map) File "/usr/local/lib/python3.10/site-packages/torch/utils/data/_utils/collate.py", line 240, in collate raise TypeError(default_collate_err_msg_format.format(elem_type)) TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.PngImagePlugin.PngImageFile'>

Container logs: