diff options
Diffstat (limited to 'candle-examples/examples/yolo-v8/model.rs')
| -rw-r--r-- | candle-examples/examples/yolo-v8/model.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-examples/examples/yolo-v8/model.rs b/candle-examples/examples/yolo-v8/model.rs index 98a0cb63..d7fe5c12 100644 --- a/candle-examples/examples/yolo-v8/model.rs +++ b/candle-examples/examples/yolo-v8/model.rs @@ -93,6 +93,7 @@ impl ConvBlock { padding, stride, groups: 1, + dilation: 1, }; let conv = conv2d_no_bias(c1, c2, k, cfg, vb.pp("conv"))?; let bn = batch_norm(c2, 1e-3, vb.pp("bn"))?; |