diff options
Diffstat (limited to 'candle-wasm-examples/yolo/src/model.rs')
| -rw-r--r-- | candle-wasm-examples/yolo/src/model.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-wasm-examples/yolo/src/model.rs b/candle-wasm-examples/yolo/src/model.rs index a63c6e94..e0fa7ac4 100644 --- a/candle-wasm-examples/yolo/src/model.rs +++ b/candle-wasm-examples/yolo/src/model.rs @@ -97,6 +97,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"))?; |