It won’t work because later code still requires it in that format. The summary is that I need to find out why the DataLoader keeps collating my batch into one big dictionary where the keys have a list of the batches values instead of a list of dictionaries, one per sample.
My getitem is returning image, mydict but when the DataLoader gets a batch it becomes [image1, image2], {key1: [target1, target2], key2: [target1, target2]}