Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 188 Bytes
Newer Older
nateanl's avatar
Add bucketize sampler and dataset for HuBERT Base model training pipeline (#2000)  
nateanl committed
1 
from .hubert_dataset import (
Zhaoheng Ni's avatar
[Example] Refactor BucketizeBatchSampler and HuBERTDataset (#2150)  
Zhaoheng Ni committed
2 
    BucketizeBatchSampler,
nateanl's avatar
Add bucketize sampler and dataset for HuBERT Base model training pipeline (#2000)  
nateanl committed
3
4
5
6
7
8 
    CollateFnHubert,
    HuBERTDataSet,
)


__all__ = [
Zhaoheng Ni's avatar
[Example] Refactor BucketizeBatchSampler and HuBERTDataset (#2150)  
Zhaoheng Ni committed
9 
    "BucketizeBatchSampler",
nateanl's avatar
Add bucketize sampler and dataset for HuBERT Base model training pipeline (#2000)  
nateanl committed
10
11
12 
    "CollateFnHubert",
    "HuBERTDataSet",
]