Skip to content

Commit

Permalink
dbg & ci script fix
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Dec 13, 2023
1 parent b816497 commit 2c6c232
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion examples/nnef-dump-mobilenet-v2/ci.sh
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@

set -ex

wget https://storage.googleapis.com/mobilenet_v2/checkpoints/mobilenet_v2_1.4_224.tgz
[ -e mobilenet_v2_1.4_224.tgz ] || wget https://storage.googleapis.com/mobilenet_v2/checkpoints/mobilenet_v2_1.4_224.tgz
tar zxf mobilenet_v2_1.4_224.tgz

cargo run -p tract -- mobilenet_v2_1.4_224_frozen.pb -i 1,224,224,3,f32 dump --nnef mobilenet.nnef.tgz
Expand Down
4 changes: 2 additions & 2 deletions nnef/src/ops/nnef/deser.rs
Original file line number Diff line number Diff line change
Expand Up @@ -340,12 +340,12 @@ pub fn conv_or_deconv(
}

let mut inputs = tvec!(input, kernel, bias);
let (group, pool_spec) = dbg!(read_conv_parameters(
let (group, pool_spec) = read_conv_parameters(
builder,
invocation,
kernel_fact.shape.as_concrete().context("Except fixed kernel shape")?,
&input_fact,
)?);
)?;

let output_dt: Option<DatumType> = if input_fact.datum_type.is_float() {
None
Expand Down

0 comments on commit 2c6c232

Please sign in to comment.