trinity.manager.config_registry.buffer_config_manager module

trinity.manager.config_registry.buffer_config_manager module#

trinity.manager.config_registry.buffer_config_manager.set_total_epochs(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_total_steps(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_explore_batch_size(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.get_train_batch_size() → int[source]#

trinity.manager.config_registry.buffer_config_manager.get_train_batch_size_per_gpu() → int[source]#

trinity.manager.config_registry.buffer_config_manager.set_train_batch_size(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.check_train_batch_size(unfinished_fields: set, *, key: str = 'train_batch_size')#

trinity.manager.config_registry.buffer_config_manager.set_buffer_max_retry_times(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_max_retry_interval(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_taskset_path(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.check_taskset_path(unfinished_fields: set, *, key: str = 'taskset_path')#

trinity.manager.config_registry.buffer_config_manager.set_taskset_args(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_eval_tasksets(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_default_workflow_type(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_default_eval_workflow_type(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_default_reward_fn_type(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_storage_type(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_enable_replay_buffer(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_reuse_cooldown_time(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_priority_fn(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.parse_priority_fn_args(raw_data: str)[source]#

trinity.manager.config_registry.buffer_config_manager.set_priority_fn_args(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_experience_buffer_path(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.check_experience_buffer_path(unfinished_fields: set, *, key: str = 'experience_buffer_path')#

trinity.manager.config_registry.buffer_config_manager.set_dpo_dataset_kwargs(**kwargs)#

trinity.manager.config_registry.buffer_config_manager.set_sft_dataset_kwargs(**kwargs)#