|
0.13.2
| 2024 年 5 月 3 日 |
|---|
|
0.12.1
| 2024 年 2 月 1 日 |
|---|
|
0.11.1
| 2023 年 12 月 4 日 |
|---|
|
0.10.0
| 2023 年 10 月 24 日 |
|---|
|
0.8.0
| 2023 年 7 月 25 日 |
|---|
#935 in 机器学习
每月 347 次下载
在 fibertools-rs 中使用
导入模型
Burn 项目支持从各种框架导入模型，强调效率和兼容性。目前，它处理两种主要的模型格式
-
ONNX：促进直接导入，确保模型性能和结构得到保持。
-
PyTorch：使 PyTorch 模型权重能够加载到 Burn 的原生模型架构中，确保无缝集成。
贡献
对 burn-import 的贡献感兴趣？查看我们的 开发指南 获取更多信息。
burn-import 是一个 crate，旨在简化将其他机器学习框架中训练的模型导入 Burn 框架的过程。此工具生成一个 Rust 源文件，使导入的模型与 Burn 的模型保持一致，并将张量数据转换为与 Burn 兼容的格式。