Skip to content
Snippets Groups Projects

Use the ordered nodes API in Sequential() in order to preserve the user declaration nodes order

Merged Olivier BICHLER requested to merge seq_with_nodes_order into dev
Files
2
+ 2
2
@@ -42,8 +42,8 @@ Aidge::DataProvider::DataProvider(const Aidge::Database& database, const std::si
@@ -42,8 +42,8 @@ Aidge::DataProvider::DataProvider(const Aidge::Database& database, const std::si
// Compute the number of bacthes depending on mDropLast boolean
// Compute the number of bacthes depending on mDropLast boolean
mNbBatch = (mDropLast) ?
mNbBatch = (mDropLast) ?
static_cast<std::size_t>(std::floor(mNbItems / mBatchSize)) :
(mNbItems / mBatchSize) :
static_cast<std::size_t>(std::ceil(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
std::vector<std::shared_ptr<Aidge::Tensor>> Aidge::DataProvider::readBatch() const
Loading