From 7f85a75cfa51a28b96323b85835e257603b4f257 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Fri, 31 May 2024 17:55:54 +0200
Subject: [PATCH] Fixed bug with dropLast
---
src/data/DataProvider.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/data/DataProvider.cpp b/src/data/DataProvider.cpp
index 5c3d1d7ef..fc6b842ed 100644
--- a/src/data/DataProvider.cpp
+++ b/src/data/DataProvider.cpp
@@ -42,8 +42,8 @@ Aidge::DataProvider::DataProvider(const Aidge::Database& database, const std::si
// Compute the number of bacthes depending on mDropLast boolean
mNbBatch = (mDropLast) ?
- static_cast<std::size_t>(std::floor(mNbItems / mBatchSize)) :
- static_cast<std::size_t>(std::ceil(mNbItems / mBatchSize));
+ (mNbItems / mBatchSize) :
+ static_cast<std::size_t>(std::ceil(mNbItems / static_cast<float>(mBatchSize)));
}
std::vector<std::shared_ptr<Aidge::Tensor>> Aidge::DataProvider::readBatch() const
--
GitLab