Functions — github.com/OpenBMB/VisRAG

hub / github.com/OpenBMB/VisRAG / functions

Functions5,024 in github.com/OpenBMB/VisRAG

⨍ Functions 5,024◇ Types & classes 888↳ Endpoints 16

↓ 1 callersFunction_get_unpad_data

(attention_mask)

src/openmatch/modeling/weighted_selection/MiniCPMV20/modeling_minicpm.py:80

↓ 1 callersFunction_get_unpad_data

(attention_mask)

src/openmatch/modeling/modeling_minicpmv/modeling_minicpm.py:80

↓ 1 callersFunction_get_unpad_data

(attention_mask)

src/openmatch/modeling/modeling_siglip/modeling_siglip.py:67

↓ 1 callersFunction_get_weighted_transforms

(transforms: Dict)

timm_modified/timm/data/auto_augment.py:707

↓ 1 callersFunction_group

(it, size)

timm_modified/timm/optim/optim_factory.py:64

↓ 1 callersFunction_in_head

(n, hp)

timm_modified/timm/optim/optim_factory.py:70

↓ 1 callersFunction_in_projection

r""" Performs the in-projection step of the attention operation. This is simply a triple of linear projections, with shape constraints on the

src/openmatch/modeling/weighted_selection/MiniCPMV20/resampler.py:779

↓ 1 callersFunction_in_projection_packed

r""" Performs the in-projection step of the attention operation, using packed weights. Output is a triple containing projection tensors for qu

src/openmatch/modeling/weighted_selection/MiniCPMV20/resampler.py:719

↓ 1 callersMethod_init_group

(self, group, params_with_grad, d_p_list, momentum_buffer_list)

timm_modified/timm/optim/sgdw.py:54

↓ 1 callersMethod_init_respostnorm

(self)

timm_modified/timm/models/swin_transformer_v2.py:442

↓ 1 callersMethod_init_rope

(self)

src/openmatch/modeling/weighted_selection/MiniCPMV20/modeling_minicpm.py:400

↓ 1 callersMethod_init_rope

(self)

src/openmatch/modeling/modeling_minicpmv/modeling_minicpm.py:400

↓ 1 callersMethod_init_with_detached_workers

(self, worker_names: List[str])

src/rsgrpo/verl/single_controller/ray/base.py:244

↓ 1 callersMethod_init_with_resource_pool

( self, resource_pool: RayResourcePool, ray_cls_with_init: RayClassWithInitArgs, bin_pack: bool, detac

src/rsgrpo/verl/single_controller/ray/base.py:249

↓ 1 callersMethod_initialize_weights

(self)

timm_modified/timm/models/nextvit.py:525

↓ 1 callersMethod_initialize_weights

(self)

timm_modified/timm/models/vgg.py:163

↓ 1 callersMethod_intermediate_layers

( self, x: torch.Tensor, n: Union[int, Sequence] = 1, )

timm_modified/timm/models/vision_transformer.py:633

↓ 1 callersFunction_interpolation

(kwargs)

timm_modified/timm/data/auto_augment.py:53

↓ 1 callersFunction_is_contiguous

(tensor: torch.Tensor)

timm_modified/timm/layers/norm.py:77

↓ 1 callersMethod_is_worker_alive

(self, worker)

src/rsgrpo/verl/single_controller/base/worker_group.py:110

↓ 1 callersFunction_kernel_valid

(k)

timm_modified/timm/layers/selective_kernel.py:15

↓ 1 callersFunction_label_from_paths

(*path, leaf_only=True)

timm_modified/timm/data/readers/reader_image_in_tar.py:125

↓ 1 callersFunction_layer_map

(model, layers_per_group=12, num_groups=None)

timm_modified/timm/optim/optim_factory.py:69

↓ 1 callersFunction_layer_norm_cf

(x: torch.Tensor, weight: torch.Tensor, bias: torch.Tensor, eps: float)

timm_modified/timm/layers/norm.py:86

↓ 1 callersMethod_load_checkpoint

(self, checkpoint_path)

timm_modified/timm/utils/model_ema.py:52

↓ 1 callersMethod_load_checkpoint

(self)

src/rsgrpo/verl/trainer/ray_trainer.py:346

↓ 1 callersFunction_load_info

(root, names=('_info.json', 'info.json'))

timm_modified/timm/data/readers/reader_wds.py:42

↓ 1 callersFunction_load_weights

(model: nn.Module, checkpoint_path: str, prefix: str = 'resnet/')

timm_modified/timm/models/resnetv2.py:496

↓ 1 callersFunction_load_weights

Load weights from .npz checkpoints for official Google Brain Flax implementation

timm_modified/timm/models/vision_transformer.py:797

↓ 1 callersMethod_make_batch_data

(self, metrics: Dict[str, Any])

src/rsgrpo/verl/trainer/ray_trainer.py:467

↓ 1 callersMethod_make_block

(self, ba, block_idx, block_count)

timm_modified/timm/models/_efficientnet_builder.py:315

↓ 1 callersMethod_make_branches

(self, num_branches, block_type, num_blocks, num_channels)

timm_modified/timm/models/hrnet.py:422

↓ 1 callersFunction_make_divisible

(v, divisor, min_value=None)

timm_modified/timm/models/nextvit.py:95

↓ 1 callersMethod_make_fuse_layers

(self)

timm_modified/timm/models/hrnet.py:429

↓ 1 callersMethod_make_one_branch

(self, branch_index, block_type, num_blocks, num_channels, stride=1)

timm_modified/timm/models/hrnet.py:405

↓ 1 callersMethod_make_weight_iterator

( self, actor_weights: Dict[str, Union[torch.Tensor, DTensor]] )

src/rsgrpo/verl/workers/sharding_manager/fsdp_vllm.py:86

↓ 1 callersFunction_materialize_futures

(*args, **kwargs)

src/rsgrpo/verl/single_controller/base/decorator.py:182

↓ 1 callersMethod_maybe_log_val_generations

Log a table of validation samples

src/rsgrpo/verl/trainer/ray_trainer.py:375

↓ 1 callersFunction_mha_shape_check

(query: Tensor, key: Tensor, value: Tensor, key_padding_mask: Optional[Tensor], attn_mask

src/openmatch/modeling/weighted_selection/MiniCPMV20/resampler.py:635

↓ 1 callersMethod_mix_batch

(self, x)

timm_modified/timm/data/mixup.py:196

↓ 1 callersMethod_mix_batch_collate

(self, output, batch)

timm_modified/timm/data/mixup.py:280

↓ 1 callersMethod_mix_elem

(self, x)

timm_modified/timm/data/mixup.py:159

↓ 1 callersMethod_mix_elem_collate

(self, output, batch, half=False)

timm_modified/timm/data/mixup.py:227

↓ 1 callersMethod_mix_pair

(self, x)

timm_modified/timm/data/mixup.py:176

↓ 1 callersMethod_mix_pair_collate

(self, output, batch)

timm_modified/timm/data/mixup.py:252

↓ 1 callersFunction_none_or_dtype

(input: Optional[Tensor])

src/openmatch/modeling/weighted_selection/MiniCPMV20/resampler.py:712

↓ 1 callersFunction_num_groups

(num_channels, num_groups, group_size)

timm_modified/timm/layers/norm_act.py:339

↓ 1 callersMethod_optimizer_step

(self)

src/rsgrpo/verl/workers/critic/dp_critic.py:125

↓ 1 callersMethod_optimizer_step

(self)

src/rsgrpo/verl/workers/actor/dp_actor.py:157

↓ 1 callersFunction_overlay_kwargs

(cfg: MaxxVitCfg, **kwargs)

timm_modified/timm/models/maxxvit.py:1112

↓ 1 callersMethod_pad_1x1_to_3x3_tensor

(kernel1x1)

timm_modified/timm/models/repghost.py:81

↓ 1 callersMethod_padding

(self, x)

timm_modified/timm/layers/median_pool.py:25

↓ 1 callersFunction_parse_args

()

timm_modified/train.py:394

↓ 1 callersFunction_parse_split_info

(split: str, info: Dict)

timm_modified/timm/data/readers/reader_wds.py:75

↓ 1 callersMethod_partition_attn

(self, x)

timm_modified/timm/models/maxxvit.py:722

↓ 1 callersMethod_partition_attn

(self, x)

timm_modified/timm/models/maxxvit.py:797

↓ 1 callersMethod_partition_attn

(self, x)

timm_modified/timm/models/maxxvit.py:896

↓ 1 callersMethod_pos_embed

(self, x)

timm_modified/timm/models/eva.py:532

↓ 1 callersFunction_posterize_level_to_arg

(level, _hparams)

timm_modified/timm/data/auto_augment.py:256

↓ 1 callersMethod_prepare_data

(self)

src/openmatch/dataset/inference_dataset.py:111

↓ 1 callersMethod_prepare_inputs

( self, inputs: Tuple[Dict[str, Union[torch.Tensor, Any]], ...] )

src/openmatch/trainer/dense_trainer.py:107

↓ 1 callersMethod_process_list

( self, tokenizer, data_list: List[str], max_inp_length: Optional[int] = None )

src/openmatch/modeling/weighted_selection/MiniCPMV20/modeling_minicpmv.py:197

↓ 1 callersFunction_process_multi_modal_data

( multi_modal_data: Dict[str, Any], min_pixels: int, max_pixels: int, video_fps: float )

src/rsgrpo/verl/workers/rollout/vllm_rollout_spmd.py:52

↓ 1 callersMethod_relative_positional_encodings

Method computes the relative positional encodings Returns: relative_position_bias (torch.Tensor): Relative positional encodings

timm_modified/timm/models/swin_transformer_v2_cr.py:165

↓ 1 callersMethod_rename_weight_keys

(self, actor_weights: Dict[str, Union[torch.Tensor, DTensor]], model: PreTrainedModel)

src/rsgrpo/verl/workers/sharding_manager/fsdp_vllm.py:66

↓ 1 callersMethod_repeat

(self, query, N: int)

src/openmatch/modeling/weighted_selection/MiniCPMV20/resampler.py:179

↓ 1 callersMethod_repeat

(self, query, N: int)

src/openmatch/modeling/modeling_minicpmv/resampler.py:170

↓ 1 callersMethod_rope_scaling_validation

Validate the `rope_scaling` configuration.

src/openmatch/modeling/weighted_selection/MiniCPMV20/configuration_minicpm.py:170

↓ 1 callersMethod_rope_scaling_validation

Validate the `rope_scaling` configuration.

src/openmatch/modeling/modeling_minicpmv/configuration_minicpm.py:170

↓ 1 callersMethod_shifted_window_attn

(self, x)

timm_modified/timm/models/swin_transformer_v2_cr.py:337

↓ 1 callersMethod_shortcut_pool

(self, x, feat_size: List[int])

timm_modified/timm/models/mvitv2.py:589

↓ 1 callersFunction_solarize_level_to_arg

(level, _hparams)

timm_modified/timm/data/auto_augment.py:277

↓ 1 callersMethod_sync_weight_to_vllm

(self)

src/rsgrpo/verl/workers/sharding_manager/fsdp_vllm.py:92

↓ 1 callersFunction_test_state_dict

(weight, bias, input, constructor)

timm_modified/tests/test_optim.py:64

↓ 1 callersFunction_trunc_normal_

(tensor, mean, std, a, b)

src/openmatch/modeling/modeling_siglip/modeling_siglip.py:80

↓ 1 callersFunction_try_run

( model_name, bench_fn, bench_kwargs, initial_batch_size, no_batch_siz

timm_modified/benchmark.py:545

↓ 1 callersMethod_upad_input

( self, query_layer, key_layer, value_layer, attention_mask, query_length )

src/openmatch/modeling/weighted_selection/MiniCPMV20/modeling_minicpm.py:770

↓ 1 callersMethod_upad_input

( self, query_layer, key_layer, value_layer, attention_mask, query_length )

src/openmatch/modeling/modeling_minicpmv/modeling_minicpm.py:770

↓ 1 callersMethod_upad_input

(self, query_layer, key_layer, value_layer, attention_mask, query_length)

src/openmatch/modeling/modeling_siglip/modeling_siglip.py:606

↓ 1 callersFunction_update_default_model_kwargs

Update the default_cfg and kwargs before passing to model Args: pretrained_cfg: input pretrained cfg (updated in-place) kwargs:

timm_modified/timm/models/_builder.py:264

↓ 1 callersFunction_weight_init

(m)

timm_modified/timm/models/senet.py:29

↓ 1 callersMethod_window_attn

(self, x, q_global: Optional[torch.Tensor] = None)

timm_modified/timm/models/gcvit.py:278

↓ 1 callersFunction_xception

(variant, pretrained=False, **kwargs)

timm_modified/timm/models/xception.py:224

↓ 1 callersFunctionaccuracy

Computes the accuracy over the k top predictions for the specified values of k

timm_modified/timm/utils/metrics.py:25

↓ 1 callersFunctionaccuracy_np

(output, target)

timm_modified/onnx_validate.py:102

↓ 1 callersFunctionaccuracy_reward

(response: str, ground_truth: str)

src/rsgrpo/examples/reward_function/math.py:27

↓ 1 callersFunctionaccuracy_reward

(response: str, ground_truth: str)

src/rsgrpo/examples/reward_function/dapo.py:119

↓ 1 callersFunctionaccuracy_reward

(response: str, ground_truth: str)

src/rsgrpo/examples/reward_function/r1v.py:27

↓ 1 callersFunctionact_with_gamma

(act_type, gamma: float = 1.)

timm_modified/timm/models/nfnet.py:74

↓ 1 callersFunctionadapt_model_from_file

(parent_module, model_variant)

timm_modified/timm/models/_prune.py:111

↓ 1 callersFunctionadapt_model_from_string

(parent_module, model_string)

timm_modified/timm/models/_prune.py:54

↓ 1 callersFunctionadaptive_clip_grad

(parameters, clip_factor=0.01, eps=1e-3, norm_type=2.0)

timm_modified/timm/utils/agc.py:30

↓ 1 callersFunctionadaptive_pool_feat_mult

(pool_type='avg')

timm_modified/timm/layers/adaptive_avgmax_pool.py:23

↓ 1 callersFunctionadd_pdfs

(pdf_dir)

visrag_scripts/demo/visrag_pipeline/build_index.py:14

↓ 1 callersMethodadd_result

(self, output)

timm_modified/timm/data/real_labels.py:30

↓ 1 callersFunctionadjust_widths_groups_comp

Adjusts the compatibility of widths and groups.

timm_modified/timm/models/regnet.py:69

↓ 1 callersFunctionall_gather_tensor

(local_tensor: Tensor, group: Optional[dist.ProcessGroup] = None, async_op: bool = False)

src/rsgrpo/verl/utils/ulysses.py:156

↓ 1 callersFunctionallgather_dict_tensors

TODO: optimize this. - We can use async ops - We can use only one allgather

src/rsgrpo/verl/protocol.py:706

↓ 1 callersFunctionanswer_question

(images, question)

visrag_scripts/demo/visrag_pipeline/answer.py:46

↓ 1 callersMethodany_requires_grad

(self, x)

timm_modified/timm/models/densenet.py:50

← previous next →901–1,000 of 5,024, ranked by callers