docs

游雁

2024-05-30 6d3a3da8a8c7d1be9740a9b2d6fac767f8dfff17

 funasr/datasets/large_datasets/datapipes/filter.py

@@ -1,14 +1,13 @@
from torch.utils.data import IterableDataset


def default_fn(data):
    return data


class FilterIterDataPipe(IterableDataset):

    def __init__(self,
                 datapipe,
                 fn=default_fn):
    def __init__(self, datapipe, fn=default_fn):
        self.datapipe = datapipe
        self.fn = fn