Dataset/ Data loader
Last updated
Last updated
collate_fn: It takes list of samples, where you get each sample form __getitem__() func of your dataset. So it this function basically get list of samples, where n is your batch_size. And turn it into batch sample.