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。

调用示例

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