CPM-9G-8B/9G-Train/cpm/cpm9g/training_tasks/__init__.py

3 lines
73 B
Python

from .pretrain import MixedDataset
from .finetune import FinetuneDataset