Commit 39ac0c4d authored by remi cresson's avatar remi cresson

Merge branch 'develop'

parents dadfc47f f18f4184
......@@ -187,7 +187,7 @@
same "printed page" as the copyright notice for easier
identification within third-party archives.
Copyright [yyyy] [name of copyright owner]
Copyright [2018] [Rémi Cresson (IRSTEA)]
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
......
......@@ -138,7 +138,7 @@ TensorflowMultisourceModelLearningBase<TInputImage>
// Batch start and size
const IndexValueType sampleStart = batch * m_BatchSize;
IndexValueType batchSize = m_BatchSize;
if (rest != 0)
if (rest != 0 && batch == nBatches - 1)
{
batchSize = rest;
}
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment