Skip to content

components finetune_submit

github-actions[bot] edited this page Apr 3, 2024 · 3 revisions

Submit AOAI FT Job

finetune_submit

Overview

Component to submit FT job to Azure OpenAI resource

Version: 0.0.3

View in Studio: https://ml.azure.com/registries/azureml/components/finetune_submit/version/0.0.3

Inputs

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
model GPT model engine string gpt-35-turbo-0613 False
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
data_upload_output Contains data upload output file in JSON/custom class format uri_file False

Outputs

Name Description Type
finetune_submit_output Contains finetune submit output file in JSON/custom class format uri_file

Environment

azureml://registries/azureml-staging/environments/aoai-data-upload-finetune/versions/1

Clone this wiki locally