Function collate_wrapper

data/dataset_base.py:790–793

()

def collate_wrapper():
    def collate_fn(batch):
        return SimpleCustomBatch(batch)
    return collate_fn