######

jamjamjon · Dec 29, 2024 · a30adb7 · a30adb7
1 parent cfb1a06
commit a30adb7
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 8 deletions.
diff --git a/src/misc/ops.rs b/src/misc/ops.rs
@@ -181,7 +181,8 @@ impl Ops<'_> {
     }
 
     pub fn make_divisible(x: usize, divisor: usize) -> usize {
-        (x + divisor - 1) / divisor * divisor
+        // (x + divisor - 1) / divisor * divisor
+        x.div_ceil(divisor)
     }
 
     // deprecated

diff --git a/src/models/yolo/preds.rs b/src/models/yolo/preds.rs
@@ -227,13 +227,13 @@ impl YOLOPredsFormat {
         x: ArrayBase<ViewRepr<&'a f32>, Dim<IxDynImpl>>,
         nc: usize,
     ) -> (
-        Option<ArrayView<f32, IxDyn>>,
-        Option<ArrayView<f32, IxDyn>>,
-        ArrayView<f32, IxDyn>,
-        Option<ArrayView<f32, IxDyn>>,
-        Option<ArrayView<f32, IxDyn>>,
-        Option<ArrayView<f32, IxDyn>>,
-        Option<ArrayView<f32, IxDyn>>,
+        Option<ArrayView<'a, f32, IxDyn>>,
+        Option<ArrayView<'a, f32, IxDyn>>,
+        ArrayView<'a, f32, IxDyn>,
+        Option<ArrayView<'a, f32, IxDyn>>,
+        Option<ArrayView<'a, f32, IxDyn>>,
+        Option<ArrayView<'a, f32, IxDyn>>,
+        Option<ArrayView<'a, f32, IxDyn>>,
     ) {
         match self.task() {
             Task::ImageClassification => (None, None, x, None, None, None, None),