Dataset/ Data loader

  • 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 nnsamples, where n is your batch_size. And turn it into batch sample.

Last updated