Cached dataset

Classes to create a dataset with cached labels.

source

CachedOpenCLIPDatasetConfig

 CachedOpenCLIPDatasetConfig (store_dict:dict, dataset_to_gpu:bool)

source

CachedOpenCLIPDataset

 CachedOpenCLIPDataset (device:torch.device=device(type='cpu'),
                        save_type=None, **parameters)

*Adds .caching to the ConfigDataset class.

Cached dataset that caches the label y prompts using the CLIP text_encoder. This speeds up training significantly.*

Back to top