ding.utils.data

ding.utils.data

Full Source Code

../ding/utils/data/__init__.py

1from .collate_fn import diff_shape_collate, default_collate, default_decollate, timestep_collate, ttorch_collate 2from .dataloader import AsyncDataLoader 3from .dataset import NaiveRLDataset, D4RLDataset, HDF5Dataset, BCODataset, \ 4 create_dataset, hdf5_save, offline_data_save_type 5from .rlhf_online_dataset import OnlineRLDataset 6from .rlhf_offline_dataset import OfflineRLDataset