diff options
Diffstat (limited to 'candle-examples/examples/yolo-v8/main.rs')
| -rw-r--r-- | candle-examples/examples/yolo-v8/main.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-examples/examples/yolo-v8/main.rs b/candle-examples/examples/yolo-v8/main.rs index 616e04ed..3b9c1ce9 100644 --- a/candle-examples/examples/yolo-v8/main.rs +++ b/candle-examples/examples/yolo-v8/main.rs @@ -101,7 +101,11 @@ impl ConvBlock { padding: Option<usize>, ) -> Result<Self> { let padding = padding.unwrap_or(k / 2); - let cfg = Conv2dConfig { padding, stride }; + let cfg = Conv2dConfig { + padding, + stride, + groups: 1, + }; let conv = conv2d_no_bias(c1, c2, k, cfg, vb.pp("conv"))?; let bn = batch_norm(c2, 1e-3, vb.pp("bn"))?; Ok(Self { conv, bn }) |