components aoai_finetune_pipeline - Azure/azureml-assets GitHub Wiki
Pipeline component for proxy fine-tuning with AOAI
Version: 0.0.1
View in Studio: https://ml.azure.com/registries/azureml/components/aoai_finetune_pipeline/version/0.0.1
| Name | Description | Type | Default | Optional | Enum |
|---|---|---|---|---|---|
| endpoint_name | The endpoint name or AOAI resource name. | string | False | ||
| endpoint_resource_group | Resource group for the AOAI resource. | string | True | ||
| endpoint_subscription | Subscription for the AOAI resource. | string | True | ||
| train_dataset | source file/folder for training dataset. | uri_file | False | ||
| validation_dataset | source file/folder for validation dataset. | uri_file | True | ||
| model | GPT model engine | string | gpt-35-turbo-0613 | False | ['babbage-002', 'davinci-002', 'gpt-35-turbo-0125', 'gpt-35-turbo-1106', 'gpt-35-turbo-0613', 'gpt-4'] |
| task_type | Dataset type - chat or completion | string | False | ['chat', 'completion'] | |
| registered_model_name | User-defined registered model name | string | False | ||
| n_epochs | Number of training epochs. If set to -1, number of epochs will be determined dynamically based on the input data. | integer | -1 | False | |
| batch_size | Global batch size. If set to -1, batch size will be determined dynamically based on the input data. | integer | -1 | False | |
| learning_rate_multiplier | The learning rate multiplier to use for training. | number | 1.0 | False |
| Name | Description | Type |
|---|---|---|
| finetune_submit_output | Contains finetune submit output file in JSON/custom class format | uri_file |