petastorm
petastorm copied to clipboard
Can we input a custom collate function as an input variable when creating the dataloader ?
Similar to the transform_spec function, is there a way to input a custom collator function?