Trouble in iterating dataset


_MapDatasetFetcher.fetch(self, possibly_batched_index)

The issue seems to be within the collate_fn. What is the collate_fn that you are passing into DataLoader? Within that function, is F defined?