eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: squeezenet graph metadata

Browse Source 
run *all* tests before you push ‼️
This commit is contained in:
Carson M.
2023-12-27 23:59:24 -06:00
parent 153d7afdb9
commit 5b5c61ea30
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/squeezenet.rs
View File

@@ -23,8 +23,8 @@ fn squeezenet_mushroom() -> ort::Result<()> {
.expect("Could not download model from file");
let metadata = session.metadata()?;
assert_eq!(metadata.name()?, "main");
assert_eq!(metadata.producer()?, "");
assert_eq!(metadata.name()?, "main_graph");
assert_eq!(metadata.producer()?, "pytorch");
let class_labels = get_imagenet_labels()?;