Method__init__(self, num_beams, max_len, min_len, evaluate, report_metric=True)
code/lavis/tasks/captioning_custom.py:22
Method__init__(
self,
num_beams,
max_len,
min_len,
evaluate,
num_ans_candida
code/lavis/tasks/vqa_reading_comprehension.py:24
Method__init__(
self,
optimizer,
max_epoch,
min_lr,
init_lr,
decay_rate=1,
code/lavis/common/optims.py:15
Method__init__(
self,
optimizer,
max_epoch,
min_lr,
init_lr,
warmup_steps=0,
code/lavis/common/optims.py:58
Method__init__(
self,
dataset,
batch_size,
num_replicas=None,
rank=None,
shu
code/lavis/datasets/datasets/hdfs_distributed_sampler.py:6
Method__init__(self, width: int, layers: int, heads: int, attn_mask: torch.Tensor = None, use_grad_checkpointing=False)
code/lavis/models/clip_vit.py:143
Method__init__(self, input_resolution: int, patch_size: int, width: int, layers: int, heads: int, use_grad_checkpointing: bo
code/lavis/models/clip_vit.py:154
Method__init__(
self, dim, num_heads=8, qkv_bias=False, qk_scale=None, attn_drop=0.,
proj_drop=0., w
code/lavis/models/eva_vit.py:65
Method__init__(self, dim, num_heads, mlp_ratio=4., qkv_bias=False, qk_scale=None, drop=0., attn_drop=0.,
dr
code/lavis/models/eva_vit.py:153
Method__init__(self, img_size=224, patch_size=16, in_chans=3, num_classes=1000, embed_dim=768, depth=12,
nu
code/lavis/models/eva_vit.py:249
Method__init__(
self,
image_encoder,
text_encoder,
text_decoder,
queue_size,
code/lavis/models/blip_models/blip_pretrain_video.py:40
Method__init__(
self,
image_encoder,
text_encoder,
num_classes,
momentum=0.995,
code/lavis/models/blip_models/blip_classification.py:30