↓ 5 callersFunctionflat_list_of_listsflatten a list of lists [[1,2], [3,4]] to [1,2,3,4]
xmodaler/functional/func_others.py:7
↓ 4 callersMethod__init__(
self,
*,
optimizer,
min_lr,
warmup_steps,
t_total,
xmodaler/lr_scheduler/warmup_lr.py:46
↓ 4 callersFunctionbuild_xmodaler_valtest_loader(datalist, *, dataset_mapper, batch_size, num_workers, multi_gpu_eval)
xmodaler/datasets/build.py:116
↓ 3 callersMethod__init__(
self,
*,
d_model: int,
d_k: int,
d_v: int,
num_head: int
xmodaler/modeling/layers/multihead_attention.py:203
↓ 2 callersMethod__init__(
self,
*,
d_model=512,
num_head=8,
num_memory=40,
d_ff=20
xmodaler/modeling/encoder/memory_augmented_encoder.py:56
↓ 2 callersMethod__init__(
self,
*,
vocab_size,
max_seq_len,
token_embed,
visual_embed,
xmodaler/modeling/meta_arch/uniter.py:146
↓ 2 callersMethod__init__(
self,
*,
embed_dim: int,
att_heads: int,
att_mid_dim: list,
xmodaler/modeling/layers/lowrank_bilinear_layers.py:14
↓ 2 callersFunctioncaption_to_mask_tokens(caption, max_seq_length, tokenizer, need_g_tokens=False, need_no_mask_tokens=False, must_mask=False)
xmodaler/functional/func_pretrain.py:69