trinity.manager.config_registry.model_config_manager module#

trinity.manager.config_registry.model_config_manager.set_total_gpu_num()[source]#
trinity.manager.config_registry.model_config_manager.set_trainer_gpu_num()[source]#
trinity.manager.config_registry.model_config_manager.set_project(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_exp_name(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_checkpoint_root_dir(**kwargs)#
trinity.manager.config_registry.model_config_manager.check_checkpoint_root_dir(unfinished_fields: set, *, key: str = 'checkpoint_root_dir')#
trinity.manager.config_registry.model_config_manager.set_monitor_type(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_model_path(**kwargs)#
trinity.manager.config_registry.model_config_manager.check_model_path(unfinished_fields: set, *, key: str = 'model_path')#
trinity.manager.config_registry.model_config_manager.set_critic_model_path(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_max_prompt_tokens(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_min_response_tokens(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_max_response_tokens(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_max_model_len(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_node_num(**kwargs)#
trinity.manager.config_registry.model_config_manager.set_gpu_per_node(**kwargs)#