I have an Image databunch generated using the following code
data = (SegmentationItemList.from_folder(image_dir)
.split_by_rand_pct()
.label_from_func(get_y_fn, classes=codes)
.transform(get_transforms(), tfm_y=True, size=256)
.databunch(bs=9, path=data_loc)
.normalize(imagenet_stats))
When I try to get a batch from this DataBunch using data.one_batch()
, I get a the following RuntimeError
RuntimeError: DataLoader worker (pid 2813) is killed by signal: Unknown signal: 0.
During handling of the above exception, another exception occurred:
RuntimeError: DataLoader worker (pid(s) 2813) exited unexpectedly
I thought this was because the process was using too much memory. So, I reduced the image size to 128 and batch size to 4 but still faced the same problem.
How do I fix this?