esp-idf-triton

为在嵌入式系统上运行预编译模型而设计的，基于 Triton crate 的轻量级分支

esp-idf-triton 可以用于运行由 主 Triton crate 创建的预编译 ML 模型。在 Triton 程序中创建模型并使用 network.serialize_triton_fmt("out.triton"); 将创建一个与 esp-idf-triton 兼容的反序列化 .triton 文件。

即将提供文件 I/O 支持，但作为一个快速的概念证明，esp-idf-triton 目前可以解析从 .triton 文件中复制的神经网络字符串并执行前向预测。

示例

let model_str = "D|3|2|10.845654 11.002682 -13.501029 -14.699452 -53.440483 -53.715294|
    -6.101849 49.06853 61.28852#D|1|3|30.350481 -78.40228 70.861206|-19.532055#D|1|1|15.161753|-3.7315714".to_string();

let mut xor_net = Network::deserialize_triton_fmt_string(model_str);

创建了一个基于训练的 SIGMOID XoR 预测器的简单网络。

当前 .triton 文件格式默认为 SIGMOID 激活函数，但实现此文件格式中的激活函数支持是我们的首要任务。

如果您喜欢开源开发，我们希望 Triton 可以实施任何额外的工作，如果您想帮忙，请联系 [email protected]！

许可证

根据您的选择，许可协议为 Apache 许可协议第 2.0 版 http://www.apache.org/licenses/LICENSE-2.0 或 MIT 许可证 http://opensource.org/licenses/MIT。此文件不得复制、修改或分发，除非根据这些条款。