trinity.manager.config_registry package

Submodules

• trinity.manager.config_registry.algorithm_config_manager module
  • set_algorithm_type()
  • set_repeat_times()
  • set_sample_strategy()
  • set_expert_data_ratio_in_sample_strategy()
  • set_advantage_fn()
  • set_gamma_in_advantage_fn()
  • set_lam_in_advantage_fn()
  • set_epsilon_in_advantage_fn()
  • set_opmd_baseline_in_advantage_fn()
  • set_tau_in_advantage_fn()
  • set_kl_loss_fn()
  • set_kl_coef_in_kl_loss_fn()
  • set_kl_penalty_fn()
  • set_adaptive_in_kl_penalty_fn()
  • set_kl_coef_in_kl_penalty_fn()
  • set_policy_loss_fn()
  • set_clip_range_in_policy_loss_fn()
  • set_sft_loss_agg_mode()
  • set_beta_in_policy_loss_fn()
  • set_label_smoothing_in_policy_loss_fn()
  • set_tau_in_policy_loss_fn()
  • set_mu_in_policy_loss_fn()
  • set_entropy_loss_fn()
  • set_entropy_coef_in_entropy_loss_fn()
• trinity.manager.config_registry.buffer_config_manager module
  • set_total_epochs()
  • set_total_steps()
  • set_explore_batch_size()
  • get_train_batch_size()
  • get_train_batch_size_per_gpu()
  • set_train_batch_size()
  • check_train_batch_size()
  • set_buffer_max_retry_times()
  • set_max_retry_interval()
  • set_taskset_path()
  • check_taskset_path()
  • set_taskset_args()
  • set_eval_tasksets()
  • set_default_workflow_type()
  • set_default_eval_workflow_type()
  • set_default_reward_fn_type()
  • set_storage_type()
  • set_enable_replay_buffer()
  • set_reuse_cooldown_time()
  • set_priority_fn()
  • parse_priority_fn_args()
  • set_priority_fn_args()
  • set_experience_buffer_path()
  • check_experience_buffer_path()
  • set_dpo_dataset_kwargs()
  • set_sft_dataset_kwargs()
• trinity.manager.config_registry.config_registry module
  • ConfigRegistry
    • ConfigRegistry.__init__()
    • ConfigRegistry.set_unfinished_fields()
    • ConfigRegistry.default_config
    • ConfigRegistry.get()
    • ConfigRegistry.get_check_func()
    • ConfigRegistry.get_configs()
    • ConfigRegistry.register_config()
    • ConfigRegistry.register_check()
• trinity.manager.config_registry.explorer_config_manager module
  • explorer_visible()
  • set_runner_per_model()
  • set_eval_interval()
  • set_engine_num()
  • set_tensor_parallel_size()
  • check_tensor_parallel_size()
  • set_enforce_eager()
  • set_enable_prefix_caching()
  • set_enable_chunked_prefill()
  • set_gpu_memory_utilization()
  • set_dtype()
  • set_seed()
  • set_enable_thinking()
  • set_enable_history()
  • set_enable_openai_api()
  • set_enable_auto_tool_choice()
  • set_tool_call_parser()
  • set_reasoning_parser()
  • set_auxiliary_models()
  • check_auxiliary_models()
  • set_sync_method()
  • set_sync_style()
  • set_sync_interval()
  • set_sync_timeout()
• trinity.manager.config_registry.model_config_manager module
  • set_total_gpu_num()
  • set_trainer_gpu_num()
  • set_project()
  • set_exp_name()
  • set_checkpoint_root_dir()
  • check_checkpoint_root_dir()
  • set_monitor_type()
  • set_log_level()
  • set_model_path()
  • check_model_path()
  • set_critic_model_path()
  • set_max_prompt_tokens()
  • set_min_response_tokens()
  • set_max_response_tokens()
  • set_max_model_len()
  • set_node_num()
  • set_gpu_per_node()
• trinity.manager.config_registry.trainer_config_manager module
  • set_trainer_gpu_num_display()
  • use_critic()
  • set_trainer_type()
  • set_save_interval()
  • set_enable_preview()
  • set_actor_grad_clip()
  • set_training_args()
  • set_ppo_epochs()
  • set_training_strategy()
  • set_param_offload()
  • set_grad_offload()
  • set_optimizer_offload()
  • set_forward_prefetch()
  • set_offload_policy()
  • set_reshard_after_forward()
  • set_tensor_model_parallel_size()
  • set_pipeline_model_parallel_size()
  • set_virtual_pipeline_model_parallel_size()
  • set_expert_model_parallel_size()
  • set_expert_tensor_parallel_size()
  • set_context_parallel_size()
  • set_sequence_parallel()
  • set_use_distributed_optimizer()
  • set_use_dist_checkpointing()
  • set_dist_checkpointing_path()
  • set_use_mbridge()
  • set_recompute_granularity()
  • set_recompute_modules()
  • set_recompute_method()
  • set_recompute_num_layers()
  • set_resume_mode()
  • set_resume_from_path()
  • check_resume_from_path()
  • set_impl_backend()
  • set_critic_warmup()
  • set_default_hdfs_dir()
  • set_del_local_ckpt_after_load()
  • set_max_actor_ckpt_to_keep()
  • set_max_critic_ckpt_to_keep()
  • set_norm_adv_by_std_in_grpo()
  • set_use_kl_in_reward()
  • set_kl_penalty()
  • set_kl_ctrl_type()
  • set_kl_ctrl_coef()
  • set_horizon()
  • set_target_kl()
  • set_actor_ppo_micro_batch_size_per_gpu()
  • set_ref_log_prob_micro_batch_size_per_gpu()
  • set_actor_ppo_max_token_len_per_gpu()
  • set_actor_ulysses_sequence_parallel_size()
  • set_actor_entropy_from_logits_with_chunking()
  • set_actor_entropy_checkpointing()
  • set_actor_lr()
  • set_actor_warmup_style()
  • set_actor_lr_warmup_steps_ratio()
  • set_actor_save_checkpoint()
  • set_actor_load_checkpoint()
  • set_critic_lr()
  • set_critic_warmup_style()
  • set_critic_lr_warmup_steps_ratio()
  • set_critic_grad_clip()
  • set_critic_cliprange_value()
  • set_critic_ppo_micro_batch_size_per_gpu()
  • set_critic_ulysses_sequence_parallel_size()
  • set_critic_save_checkpoint()
  • set_critic_load_checkpoint()

Module contents