set_hard_label

功能说明

KnowledgeDistillConfig类方法，配置蒸馏时student模型的loss权重、loss的index，必须调用该方法。

函数原型

set_hard_label(weight, index)

参数说明

参数名	输入/返回值	含义	使用限制
weight	输入	hard loss 的权重。	必选。数据类型：float。建议取值0-1之间。
index	输入	student模型output的索引值(index)。当student模型有多个output时，由index决定哪个output来计算loss。仅MindSpore模型需要配置，通常为0。	必选。数据类型：int。

参数名

输入/返回值

含义

使用限制

weight

输入

hard loss 的权重。

必选。

数据类型：float。

建议取值0-1之间。

index

输入

student模型output的索引值(index)。当student模型有多个output时，由index决定哪个output来计算loss。

仅MindSpore模型需要配置，通常为0。

必选。

数据类型：int。

调用示例

from msmodelslim.common.knowledge_distill.knowledge_distill import KnowledgeDistillConfig
distill_config = KnowledgeDistillConfig()
distill_config.set_hard_label (0.5, 0)

父主题： KnowledgeDistillConfig