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 nnnsamples, where n is your batch_size. And turn it into batch sample.
Last updated 5 years ago