Pytorch datasets

ImageDataBunch ( train_dl : DataLoader , valid_dl : DataLoader , test_dl : Optional [ DataLoader ]= None ,

pass can we pass in our pytorch custom data loaders in to this? Can we pass in pytorch’s own inbuilt datasets into this?

You can pass any pytorch dataloader yes. If you want to pass datasets, use ImageDataBunch.create.

1 Like