-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Closed
Labels
bugSomething isn't workingSomething isn't working
Milestone
Description
Describe the bug
Currently, the ThreadDataLoader can only run for 1 epoch, it doesn't contain the reset logic.
Reproduce the issue:
dataset = Dataset(data=self.datalist, transform=self.transform)
dataloader = ThreadDataLoader(dataset=dataset, batch_size=2, num_workers=0)
for d in dataloader:
self.assertEqual(d["image"][0], "spleen_19.nii.gz")
self.assertEqual(d["image"][1], "spleen_31.nii.gz")
self.assertEqual(d["label"][0], "spleen_label_19.nii.gz")
self.assertEqual(d["label"][1], "spleen_label_31.nii.gz")
for d in dataloader:
print("will not print here.")Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working