How to create a dataloader with variable-size input

vision
2

you need to write a custom collate_fn and pass it to your data loader. Questions about Dataloader and Dataset

11 Likes