collate_fn

collate_fn is a function in PyTorch that customizes how data is batched together from a Dataset before being fed into a model.