↓ 2 callersMethodtokenize(self, texts: Union[str, List[str]], context_length: int = 77, return_length: bool = False, mask_type=None)
prototype/model/text_encoder/text_transformer.py:144
↓ 1 callersMethod__init__(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None, checkpoint: bool = False, dropout:
prototype/model/text_encoder/base_transformer.py:57
↓ 1 callersMethod__init__(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None, checkpoint: bool = False, dropout:
prototype/model/image_encoder/base_transformer.py:57
↓ 1 callersMethod__init__(self, batch_size, num_threads, device_id, crop, size)
prototype/data/pipelines/imagenet_pipeline.py:57
↓ 1 callersMethod__init__(self, root_dir, meta_file, transform=None,
read_from='mc', evaluator=None, image_reader_type
prototype/data/datasets/clip_dataset.py:40
↓ 1 callersMethod__init__(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8,
weight_decay=0, amsgrad=False, max_norm
prototype/optimizer/adam_clip.py:7