-
Notifications
You must be signed in to change notification settings - Fork 21
loss
desperadoccy edited this page May 28, 2024
·
3 revisions
loss支持加载第三方实现,可以以如下方式配置:
"loss": "torch.nn.functional.cross_entropy",亦可以加载自己实现的模块,加载方式为本框架通用的加载方式,即:
"path": "该loss类的路径",
"params": "初始化该类所需的params"
"type": "func | class"loss支持自定义,但是需要在lossfactory中进行注册
如果type为func,lossfactory创建时返回函数,如果type为class,lossfactory会返回对象。默认为func。
该类为工厂类,用于生成loss函数或者类。如配置的"loss": "torch.nn.functional.cross_entropy"之类,将会生成torch.nn.functional.cross_entropy类,否则根据path字段生成对应的类。
Federated Learning with Label Distribution Skew via Logits Calibration FedLC算法的损失函数,对客户端做数据异构修正
Getting Started - 整体流程 - Module Guide - 现有算法 - Contact Us