Cached dataset

Classes to create a dataset with cached labels.

---

source

CachedOpenCLIPDatasetConfig

def CachedOpenCLIPDatasetConfig(
    store_dict:dict, dataset_to_gpu:bool
)->None:

---

source

CachedOpenCLIPDataset

def CachedOpenCLIPDataset(
    device:device=device(type='cpu'), save_type:NoneType=None, parameters:VAR_KEYWORD
)->None:

Adds .caching to the ConfigDataset class.

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