Tuning
- Tuning settings types
- class yandex_cloud_ml_sdk._types.tuning.tuning_types.BaseTuningType
- class yandex_cloud_ml_sdk._types.tuning.tuning_types.TuningTypeLora
- class yandex_cloud_ml_sdk._types.tuning.tuning_types.TuningTypePromptTune
- class yandex_cloud_ml_sdk._types.tuning.schedulers.BaseScheduler
- class yandex_cloud_ml_sdk._types.tuning.schedulers.SchedulerLinear
- class yandex_cloud_ml_sdk._types.tuning.schedulers.SchedulerConstant
- class yandex_cloud_ml_sdk._types.tuning.schedulers.SchedulerCosine
- class yandex_cloud_ml_sdk._types.tuning.optimizers.BaseOptimizer
- class yandex_cloud_ml_sdk._types.tuning.optimizers.OptimizerAdamw
- Tuning task operation
- class yandex_cloud_ml_sdk._tuning.tuning_task.AsyncTuningTask
- class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTask
- class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTaskInfo
- class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTaskStatus
- class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTaskStatusEnum
Tuning settings types
class yandex_cloud_ml_sdk._types.tuning.tuning_types.BaseTuningType
class yandex_cloud_ml_sdk._types.tuning.tuning_types.TuningTypeLora
TuningTypeLora(rank: ‘int | None’ = None, alpha: ‘float | None’ = None, initialization: ‘str | None’ = None, type: ‘str | None’ = None)
initialization: str
class yandex_cloud_ml_sdk._types.tuning.tuning_types.TuningTypePromptTune
TuningTypePromptTune(virtual_tokens: ‘int | None’ = None)
virtual_tokens: int
class yandex_cloud_ml_sdk._types.tuning.schedulers.BaseScheduler
BaseScheduler(warmup_ratio: ‘float | None’ = None)
warmup_ratio: float
class yandex_cloud_ml_sdk._types.tuning.schedulers.SchedulerLinear
SchedulerLinear(warmup_ratio: ‘float | None’ = None, min_lr: ‘float | None’ = None)
warmup_ratio: float
class yandex_cloud_ml_sdk._types.tuning.schedulers.SchedulerConstant
SchedulerConstant(warmup_ratio: ‘float | None’ = None)
warmup_ratio: float
class yandex_cloud_ml_sdk._types.tuning.schedulers.SchedulerCosine
SchedulerCosine(warmup_ratio: ‘float | None’ = None, min_lr: ‘float | None’ = None)
warmup_ratio: float
class yandex_cloud_ml_sdk._types.tuning.optimizers.BaseOptimizer
BaseOptimizer()
class yandex_cloud_ml_sdk._types.tuning.optimizers.OptimizerAdamw
OptimizerAdamw(beta1: ‘float | None’ = None, beta2: ‘float | None’ = None, eps: ‘float | None’ = None, weight_decay: ‘float | None’ = None)
weight_decay: float
Tuning task operation
class yandex_cloud_ml_sdk._tuning.tuning_task.AsyncTuningTask
async get_task_info(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
TuningTaskInfo | None |
async get_metrics_url(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
str |
async cancel(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
None |
async get_result(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
async get_status(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
property id
async wait(*, timeout=60, poll_timeout=None, poll_interval=None)
|
Parameters |
|
|
Return type |
class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTask
get_task_info(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
TuningTaskInfo | None |
get_metrics_url(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
str |
cancel(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
None |
get_result(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
get_status(*, timeout=60)
|
Parameters |
timeout (float |
|
Return type |
property id
wait(*, timeout=60, poll_timeout=None, poll_interval=None)
|
Parameters |
|
|
Return type |
class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTaskInfo
TuningTaskInfo(id: ‘str’, _sdk: ‘BaseSDK’, task_id: ‘str’, operation_id: ‘str’, status: ‘TuningTaskStatusEnum’, folder_id: ‘str’, created_by: ‘str’, created_at: ‘datetime’, started_at: ‘datetime | None’, finished_at: ‘datetime | None’, source_model_uri: ‘str’, target_model_uri: ‘str | None’)
task_id: str
operation_id: str
status: TuningTaskStatusEnum
folder_id: str
created_by: str
created_at: datetime
started_at: datetime | None
finished_at: datetime | None
source_model_uri: str
id: str
class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTaskStatus
property is_failed: bool
Check if operation execution failed.
property is_finished: bool
Check if operation execution finished.
property is_running: bool
Check if operation execution is still in progress.
property is_succeeded: bool
Check if operation execution completed successfully.
property status_name: str
Get operation execution status name.
done: bool
error: OperationErrorInfo | None
class yandex_cloud_ml_sdk._tuning.tuning_task.TuningTaskStatusEnum
STATUS_UNSPECIFIED = 0
CREATED = 1
PENDING = 2
IN_PROGRESS = 3
COMPLETED = 4
FAILED = 5