Skip to content
desperadoccy edited this page May 28, 2024 · 3 revisions

loss参数介绍

loss支持加载第三方实现,可以以如下方式配置:

"loss": "torch.nn.functional.cross_entropy",

亦可以加载自己实现的模块,加载方式为本框架通用的加载方式,即:

"path": "该loss类的路径",
"params": "初始化该类所需的params"
"type": "func | class"

loss支持自定义,但是需要在lossfactory中进行注册 如果type为func,lossfactory创建时返回函数,如果type为class,lossfactory会返回对象。默认为func。

现有类介绍

LossFactory

该类为工厂类,用于生成loss函数或者类。如配置的"loss": "torch.nn.functional.cross_entropy"之类,将会生成torch.nn.functional.cross_entropy类,否则根据path字段生成对应的类。

FedLC

Federated Learning with Label Distribution Skew via Logits Calibration FedLC算法的损失函数,对客户端做数据异构修正

Clone this wiki locally