一种支持任意网络拓扑的高效局部学习系统及方法

    公开(公告)号:CN118378684A

    公开(公告)日:2024-07-23

    申请号:CN202311846519.7

    申请日:2023-12-29

    Abstract: 本发明一种支持任意网络拓扑的高效局部学习系统,应用于各类局部学习任务中,由具有计算能力的电子设备实现,包括:模块层封装的模块包括:rlayer模块、datasets模块、grads模块、metrics模块、optimizer模块;算子层封装的算子类、梯度求解类、优化器类、数据集类及衡量指标函数,对应模块层封装的模块,用于提供模块层所封装的模块的接口;模块层封装的算子、数据集下载与预处理类、优化器类、梯度求导类、指标计算函数均具有输入输出接口,用户可通过rlayer模块按需创建并实例化网络,通过datasets模块下载与预处理数据集,通过optimizer模块选择相应的优化器,通过grads模块选择局部学习梯度求导类,通过metrics模块选择损失与精度计算函数,完成网络及相应超参的初始化,实现网络训练与测试。

Patent Agency Ranking