diff options
| author | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
|---|---|---|
| committer | Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com> | 2024-01-07 11:52:03 +0100 |
| commit | 6ebe04327397ebf7c9400d68c43fede705f8ce75 (patch) | |
| tree | 4e701ef55370cb6226593019f9e6daaf8ddf73b4 /candle-examples/examples/yolo-v3/darknet.rs | |
| parent | 6bf52b9fdf82ad775611e82924d73172660a605e (diff) | |
| parent | 84250bf52f58528cf59dca3b82effd9f07a13cc7 (diff) | |
| download | candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.gz candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.tar.bz2 candle-6ebe04327397ebf7c9400d68c43fede705f8ce75.zip | |
Merge branch 'main' into ivarflakstad/metal-prng
Diffstat (limited to 'candle-examples/examples/yolo-v3/darknet.rs')
| -rw-r--r-- | candle-examples/examples/yolo-v3/darknet.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/yolo-v3/darknet.rs b/candle-examples/examples/yolo-v3/darknet.rs index 6702618e..b1dd3127 100644 --- a/candle-examples/examples/yolo-v3/darknet.rs +++ b/candle-examples/examples/yolo-v3/darknet.rs @@ -147,7 +147,7 @@ fn conv(vb: VarBuilder, index: usize, p: usize, b: &Block) -> Result<(usize, Bl) let func = candle_nn::func(move |xs| { let xs = conv.forward(xs)?; let xs = match &bn { - Some(bn) => bn.forward(&xs)?, + Some(bn) => xs.apply_t(bn, false)?, None => xs, }; let xs = if leaky { |