integrations.kd.args

integrations.kd.args

Plugin args for KD support.

Classes

Name	Description
InferenceServerType	Online inferences server types to handle different request args
KDArgs	Input args for knowledge distillation.
KDTrainingArgsMixin	Additional args for KD training.

InferenceServerType

integrations.kd.args.InferenceServerType()

Online inferences server types to handle different request args

KDArgs

integrations.kd.args.KDArgs()

Input args for knowledge distillation.

KDTrainingArgsMixin

integrations.kd.args.KDTrainingArgsMixin(
    kd_ce_alpha=None,
    kd_alpha=None,
    kd_temperature=None,
    kd_beta=None,
    kd_normalize_topk=None,
)

Additional args for KD training.