diff options
Diffstat (limited to 'candle-transformers/src/models/convmixer.rs')
| -rw-r--r-- | candle-transformers/src/models/convmixer.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-transformers/src/models/convmixer.rs b/candle-transformers/src/models/convmixer.rs index 7f1b75eb..7f924794 100644 --- a/candle-transformers/src/models/convmixer.rs +++ b/candle-transformers/src/models/convmixer.rs @@ -21,8 +21,8 @@ fn conv2d_same( let module = candle_nn::func(move |xs| { let ih = xs.dim(2)?; let iw = xs.dim(3)?; - let oh = (ih + s - 1) / s; - let ow = (iw + s - 1) / s; + let oh = ih.div_ceil(s); + let ow = iw.div_ceil(s); let pad_h = usize::max((oh - 1) * s + k - ih, 0); let pad_w = usize::max((ow - 1) * s + k - iw, 0); if pad_h > 0 || pad_w > 0 { |