Mixed cached dataset
Dataset that combines and handles multiple cached datasets.
This is useful for multiple qubits. Here we also handle paddings.
MixedCachedOpenCLIPDatasetConfig
MixedCachedOpenCLIPDatasetConfig (store_dict:dict, dataset_to_gpu:bool, pad_constant:int, collate_fn:str, bucket_batch_size:int, model_scale_factor:int)
MixedCachedOpenCLIPDataset
MixedCachedOpenCLIPDataset (device:torch.device=device(type='cpu'), save_type=None, **parameters)
Dataset that uses multiple cached dataset and combines them with padding, either i) Bucket or ii) Max.