diff options
Diffstat (limited to 'candle-examples/examples/yolo-v3/darknet.rs')
| -rw-r--r-- | candle-examples/examples/yolo-v3/darknet.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-examples/examples/yolo-v3/darknet.rs b/candle-examples/examples/yolo-v3/darknet.rs index d0392308..de8fcf09 100644 --- a/candle-examples/examples/yolo-v3/darknet.rs +++ b/candle-examples/examples/yolo-v3/darknet.rs @@ -128,7 +128,11 @@ fn conv(vb: VarBuilder, index: usize, p: usize, b: &Block) -> Result<(usize, Bl) } Some(_) | None => (None, true), }; - let conv_cfg = candle_nn::Conv2dConfig { stride, padding }; + let conv_cfg = candle_nn::Conv2dConfig { + stride, + padding, + groups: 1, + }; let conv = if bias { conv2d(p, filters, size, conv_cfg, vb.pp(&format!("conv_{index}")))? } else { |