diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-29 19:12:16 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-29 19:12:16 +0100 |
| commit | 2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2 (patch) | |
| tree | d2ab2b3a5b0a08903123a9039319f30e9fe6cc07 /candle-examples/examples/yolo-v8/model.rs | |
| parent | b31d41e26a47d91d828e0c4f567f14b659775e5e (diff) | |
| download | candle-2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2.tar.gz candle-2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2.tar.bz2 candle-2d3fcad26788dff3fa73996a3cc8e5fd5382f6b2.zip | |
Simplify usage of the pool functions. (#662)
* Simplify usage of the pool functions.
* Small tweak.
* Attempt at using apply to simplify the convnet definition.
Diffstat (limited to 'candle-examples/examples/yolo-v8/model.rs')
| -rw-r--r-- | candle-examples/examples/yolo-v8/model.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/candle-examples/examples/yolo-v8/model.rs b/candle-examples/examples/yolo-v8/model.rs index d7fe5c12..b834f967 100644 --- a/candle-examples/examples/yolo-v8/model.rs +++ b/candle-examples/examples/yolo-v8/model.rs @@ -198,15 +198,15 @@ impl Module for Sppf { let xs2 = xs .pad_with_zeros(2, self.k / 2, self.k / 2)? .pad_with_zeros(3, self.k / 2, self.k / 2)? - .max_pool2d((self.k, self.k), (1, 1))?; + .max_pool2d_with_stride(self.k, 1)?; let xs3 = xs2 .pad_with_zeros(2, self.k / 2, self.k / 2)? .pad_with_zeros(3, self.k / 2, self.k / 2)? - .max_pool2d((self.k, self.k), (1, 1))?; + .max_pool2d_with_stride(self.k, 1)?; let xs4 = xs3 .pad_with_zeros(2, self.k / 2, self.k / 2)? .pad_with_zeros(3, self.k / 2, self.k / 2)? - .max_pool2d((self.k, self.k), (1, 1))?; + .max_pool2d_with_stride(self.k, 1)?; self.cv2.forward(&Tensor::cat(&[&xs, &xs2, &xs3, &xs4], 1)?) } } |