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_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)