Skip to content

Commit

Permalink
Change record type to NamedMpk to avoid overflow
Browse files Browse the repository at this point in the history
This possibly fixes this:
#3

Caused by this: tracel-ai/burn#952
  • Loading branch information
antimora committed Dec 1, 2023
1 parent 99be61c commit 8f7c85b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions squeezenet-burn/build.rs
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ use burn_import::onnx::ModelGen;

const LABEL_SOURCE_FILE: &str = "src/model/label.txt";
const LABEL_DEST_FILE: &str = "model/label.rs";
const GENERATED_MODEL_WEIGHTS_FILE: &str = "squeezenet1.mpk.gz";
const GENERATED_MODEL_WEIGHTS_FILE: &str = "squeezenet1.mpk";
const INPUT_ONNX_FILE: &str = "src/model/squeezenet1.onnx";
const OUT_DIR: &str = "model/";

Expand All @@ -31,7 +31,7 @@ fn main() {
let (record_type, embed_states) = if cfg!(feature = "weights_embedded") {
(RecordType::Bincode, true)
} else {
(RecordType::NamedMpkGz, false)
(RecordType::NamedMpk, false)
};

// Check if half precision is enabled.
Expand Down

0 comments on commit 8f7c85b

Please sign in to comment.