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