How to make Dataloader avoid collate_batch

vision
2

For variable shapes, you could implement a custom collate function as described here.

1 Like