↓ 2 callersFunctionMaskTokens(tokens, mask_type, mask_token, special_tokens=None, tokenizer_length=None, sepcial_tokens_mask=None, special_
prototype/model/utils/text_utils/mask_tokens.py:32
↓ 2 callersMethod__init__(self,image_encode, text_encode, use_allgather, nn_size=2**16, nn_topk=1, \
return_dense=Fals
prototype/model/defilip.py:150
↓ 2 callersMethod__init__(self,image_encode, text_encode, use_allgather, nn_size=2**16, nn_topk=1, \
return_dense=Fals
prototype/model/declip.py:133
↓ 2 callersMethod__init__(self, custom_pipe, batch_size, num_threads, device_id,
**kwargs)
prototype/data/pipelines/imagenet_pipeline_v2.py:27
↓ 2 callersMethod__init__(self, params, lr=required, momentum=0, dampening=0,
weight_decay=0, nesterov=False, loss_sca
prototype/optimizer/fp16_optim.py:51