MCPcopy Create free account

hub / github.com/BeastyZ/ConvSearch-R1 / functions

Functions1,207 in github.com/BeastyZ/ConvSearch-R1

⨍ Functions 1,207◇ Types & classes 166↳ Endpoints 40
Method__init__
(self, model_path)
index/dense/models.py:65
Method__init__
(self, config, reward_module: nn.Module, ref_module: nn.Module, reward_optimizer: optim.Optimizer)
verl/recipe/prime/prime_dp_rm.py:43
Method__init__
(self, config)
verl/recipe/prime/prime_fsdp_workers.py:49
Method__init__
(self, config, tokenizer, role_worker_mapping: dict[Role, W
verl/recipe/prime/prime_ray_trainer.py:156
Method__init__
(self)
verl/tests/ray/test_data_transfer.py:39
Method__init__
(self, rank, world_size, group_name)
verl/tests/ray/test_rvdz.py:21
Method__init__
(self, cuda_visible_devices=None)
verl/tests/ray/test_high_level_scheduling_api.py:26
Method__init__
(self, size)
verl/tests/ray/test_worker_group_torch.py:52
Method__init__
(self, config)
verl/tests/ray/test_colocated_workers.py:39
Method__init__
(self)
verl/tests/ray/test_driverfunc_to_worker.py:32
Method__init__
(self)
verl/tests/ray/test_driverfunc_to_worker.py:38
Method__init__
(self)
verl/tests/ray/test_ray_local_envs.py:28
Method__init__
(self, x)
verl/tests/ray/test_worker_group_basics.py:44
Method__init__
(self)
verl/tests/ray/check_worker_alive/main.py:29
Method__init__
(self)
verl/tests/ray/detached_worker/server.py:51
Method__init__
Args: max_number: the maximum number allowed in the arithmetic sequence max_diff: the maximum diff. The actual commo
verl/tests/e2e/envs/digit_completion/task.py:35
Method__init__
Character tokenizer for Hugging Face transformers. Args: characters (Sequence[str]): List of desired characters. Any character wh
verl/tests/e2e/envs/digit_completion/tokenizer.py:31
Method__init__
(self, config: PretrainedConfig)
verl/verl/utils/flops_counter.py:64
Method__init__
(self, project_name, experiment_name, default_backend: Union[str, List[str]] = 'console', config=None)
verl/verl/utils/tracking.py:27
Method__init__
(self)
verl/verl/utils/tracking.py:113
Method__init__
(self)
verl/verl/utils/seqlen_balancing.py:29
Method__init__
(self, items: List[Tuple[int, int]], k: int)
verl/verl/utils/seqlen_balancing.py:51
Method__init__
(self, dictionary, **kwargs)
verl/verl/utils/py_functional.py:50
Method__init__
(self, module: nn.Module)
verl/verl/utils/memory_buffer.py:149
Method__init__
(self, transform_memory_param_fn)
verl/verl/utils/memory_buffer.py:178
Method__init__
(self, fn)
verl/verl/utils/model.py:30
Method__init__
(self, numel, numel_padded, dtype)
verl/verl/utils/megatron/memory.py:20
Method__init__
(self, remote_logger=None, enable_wandb=False, print_to_console=False)
verl/verl/utils/logger/aggregate_logger.py:32
Method__init__
(self, parquet_files: Union[str, List[str]], tokenizer, pro
verl/verl/utils/dataset/rm_dataset.py:42
Method__init__
(self, parquet_files: Union[str, List[str]], tokenizer, pro
verl/verl/utils/dataset/sft_dataset.py:39
Method__init__
(self, parquet_files: Union[str, List[str]], tokenizer: PreTrainedTokenizer,
verl/verl/utils/dataset/rl_dataset.py:80
Method__init__
(self, model: FSDP, optimizer: torch.optim.Optimizer, lr_sc
verl/verl/utils/checkpoint/fsdp_checkpoint_manager.py:47
Method__init__
(self, model: FSDP, optimizer: torch.optim.Optimizer, lr_scheduler: torch.optim.lr_scheduler.
verl/verl/utils/checkpoint/checkpoint_manager.py:42
Method__init__
(self, hdfs_dir, verbose)
verl/verl/utils/debug/trajectory_tracker.py:52
Method__init__
(self, nccl_id)
verl/verl/utils/rendezvous/ray_backend.py:27
Method__init__
(self, process_on_nodes: List[int] = None, use_gpu: bool = True,
verl/verl/single_controller/ray/base.py:71
Method__init__
(self, cls, *args, **kwargs)
verl/verl/single_controller/ray/base.py:150
Method__init__
(self, resource_pool: RayResourcePool = None, ray_cls_with_init: RayClassWit
verl/verl/single_controller/ray/base.py:198
Method__init__
(self, resource_pool: RayResourcePool, ray_cls_with_init: RayClassWithInitArgs, **kwargs)
verl/verl/single_controller/ray/megatron.py:31
Method__init__
(self, process_on_nodes=None, max_collocate_count: int = 10, n_gpus_per_node=8)
verl/verl/single_controller/base/worker_group.py:29
Method__init__
(self, cls, *args, **kwargs)
verl/verl/single_controller/base/worker_group.py:67
Method__init__
(self, resource_pool: ResourcePool, **kwargs)
verl/verl/single_controller/base/worker_group.py:95
Method__init__
(self, store)
verl/verl/single_controller/base/worker.py:73
Method__init__
(self, cuda_visible_devices=None)
verl/verl/single_controller/base/worker.py:118
Method__init__
(self, resource_pool: ResourcePool, **kwargs)
verl/verl/single_controller/base/megatron/worker_group.py:23
Method__init__
(self, cuda_visible_devices=None)
verl/verl/single_controller/base/megatron/worker.py:20
Method__init__
(self, rank_zero_info)
verl/verl/single_controller/base/register_center/ray.py:21
Method__init__
(self, config, device_mesh: DeviceMesh, ulysses_device_mesh: DeviceMesh)
verl/verl/trainer/fsdp_sft_trainer.py:80
Method__init__
(self, config, tokenizer, role_worker_mapping: dict[Role, W
verl/verl/trainer/ppo/ray_trainer.py:254
Method__init__
(self, init_kl_coef, target_kl, horizon)
verl/verl/trainer/ppo/core_algos.py:34
Method__init__
(self, kl_coef)
verl/verl/trainer/ppo/core_algos.py:49
Method__init__
(self, config: DictConfig, role: str)
verl/verl/workers/megatron_workers.py:73
Method__init__
(self, config)
verl/verl/workers/megatron_workers.py:668
Method__init__
(self, config: DictConfig, role: str)
verl/verl/workers/fsdp_workers.py:76
Method__init__
(self, config)
verl/verl/workers/fsdp_workers.py:914
Method__init__
(self, module: FSDP, inference_engine: LLM, model_config,
verl/verl/workers/sharding_manager/fsdp_vllm.py:39
Method__init__
(self, module: FSDP, inference_engine: VerlEngine, model_co
verl/verl/workers/sharding_manager/fsdp_sglang.py:49
Method__init__
(self, device_mesh: DeviceMesh)
verl/verl/workers/sharding_manager/fsdp_ulysses.py:37
Method__init__
(self, model_provider)
verl/verl/workers/sharding_manager/megatron_vllm.py:39
Method__init__
(self, module: AllGatherPPModel, inference_engine: LLM, model_config, layer_name_mapping)
verl/verl/workers/sharding_manager/megatron_vllm.py:256
Method__init__
(self, tokenizer, num_examine, compute_score=None)
verl/verl/workers/reward_manager/prime.py:89
Method__init__
(self, tokenizer, num_examine, compute_score=None)
verl/verl/workers/reward_manager/naive.py:24
Method__init__
( self, tokenizer, num_examine, compute_score, retriever_url,
verl/verl/workers/reward_manager/rewrite_r1.py:23
Method__init__
(self, config)
verl/verl/workers/critic/base.py:28
Method__init__
(self, config, model_config, megatron_config, critic_module: nn.ModuleList, critic_optimizer:
verl/verl/workers/critic/megatron_critic.py:45
Method__init__
(self, config, critic_module: nn.Module, critic_optimizer: optim.Optimizer)
verl/verl/workers/critic/dp_critic.py:41
Method__init__
Args: dataloader: an Iterable of TensorDict that consistently generates prompts. Note that the dataloader should han
verl/verl/workers/rollout/base.py:25
Method__init__
(self, module: nn.Module, config)
verl/verl/workers/rollout/hf_rollout.py:37
Method__init__
A SGLang rollout. It requires the module is supported by the SGLang. Args: actor_module: module here follows huggingface APIs
verl/verl/workers/rollout/sglang_rollout/sglang_rollout.py:86
Method__init__
A naive rollout. It requires the module to be compatible with huggingface APIs. That is: The module should define __call__ to receive input_id
verl/verl/workers/rollout/naive/naive_rollout.py:38
Method__init__
A vLLM rollout. It requires the module is supported by the vllm. Args: module: module here follows huggingface APIs c
verl/verl/workers/rollout/vllm_rollout/fire_vllm_rollout.py:60
Method__init__
A vLLM rollout. It requires the module is supported by the vllm. Args: module: module here follows huggingface APIs c
verl/verl/workers/rollout/vllm_rollout/vllm_rollout.py:59
Method__init__
A vLLM rollout. It requires the module is supported by the vllm. Args: module: module here follows huggingface APIs c
verl/verl/workers/rollout/vllm_rollout/vllm_rollout_spmd.py:67
Method__init__
MeagtronPPOActor class. This class implements the simple PPO logics when the model is built with Megatron. Args: config (OmegaCon
verl/verl/workers/actor/megatron_actor.py:55
Method__init__
The base class for PPO actor Args: config (DictConfig): a config passed to the PPOActor. We expect the type to be
verl/verl/workers/actor/base.py:28
Method__init__
When optimizer is None, it is Reference Policy
verl/verl/workers/actor/dp_actor.py:41
Method__init__
(self, config)
verl/verl/workers/reward_model/base.py:25
Method__init__
(self, config, model_config, reward_model_module: torch.nn.
verl/verl/workers/reward_model/megatron/reward_model.py:34
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig)
verl/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:159
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig)
verl/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:226
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig)
verl/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:285
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig, pre_process, post_process)
verl/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:414
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig, pre_process, post_process,
verl/verl/models/qwen2/megatron/modeling_qwen2_megatron.py:521
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig)
verl/verl/models/qwen2/megatron/layers/parallel_decoder.py:104
Method__init__
(self, config, megatron_config: ModelParallelConfig = None)
verl/verl/models/qwen2/megatron/layers/parallel_mlp.py:33
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
verl/verl/models/qwen2/megatron/layers/parallel_attention.py:75
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
verl/verl/models/qwen2/megatron/layers/parallel_attention.py:94
Method__init__
(self, config: Qwen2Config, megatron_config: ModelParallelConfig)
verl/verl/models/qwen2/megatron/layers/parallel_attention.py:146
Method__init__
(self, input_size, gate_ouput_size, up_output_size,
verl/verl/models/qwen2/megatron/layers/parallel_linear.py:54
Method__init__
Qwen2RMSNorm is equivalent to T5LayerNorm
verl/verl/models/qwen2/megatron/layers/parallel_rmsnorm.py:27
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig)
verl/verl/models/llama/megatron/modeling_llama_megatron.py:160
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig)
verl/verl/models/llama/megatron/modeling_llama_megatron.py:227
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig)
verl/verl/models/llama/megatron/modeling_llama_megatron.py:286
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig, pre_process, post_process)
verl/verl/models/llama/megatron/modeling_llama_megatron.py:415
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig, pre_process, post_process,
verl/verl/models/llama/megatron/modeling_llama_megatron.py:522
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig)
verl/verl/models/llama/megatron/layers/parallel_decoder.py:104
Method__init__
(self, config, megatron_config: ModelParallelConfig = None)
verl/verl/models/llama/megatron/layers/parallel_mlp.py:33
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
verl/verl/models/llama/megatron/layers/parallel_attention.py:75
Method__init__
(self, dim, max_position_embeddings=2048, base=10000, device=None, scaling_factor=1.0)
verl/verl/models/llama/megatron/layers/parallel_attention.py:94
Method__init__
(self, config: LlamaConfig, megatron_config: ModelParallelConfig)
verl/verl/models/llama/megatron/layers/parallel_attention.py:146
← previousnext →701–800 of 1,207, ranked by callers