dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Change ONNX default record type to NamedMpk (#1019) 

* Change ONNX default record type to NamedMpk

This temporarily workaround of #952 bug.

* Fix formatting

* Fix URL in the doc
This commit is contained in:
Dilshod Tadjibaev 2023-11-30 12:22:34 -06:00 committed by GitHub
parent 947bb60fff
commit 5e4e0b6bb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
burn-import/src/burn/graph.rs
View File

@ -21,11 +21,14 @@ pub enum RecordType {
/// Pretty JSON format (useful for debugging).
PrettyJson,
#[default]
/// Compressed Named MessagePack.
///
/// Note: This may cause infinite build.
/// See [#952 bug](https://github.com/Tracel-AI/burn/issues/952).
NamedMpkGz,
/// Uncompressed Named MessagePack.
#[default]
NamedMpk,
/// Bincode format (useful for embedding and for no-std support).