Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
Lozac'h Loic
otbtf
Commits
252f6a72
Commit
252f6a72
authored
Nov 05, 2018
by
remi cresson
Browse files
FIX: correct batch size for last iteration
parent
d2195d91
Changes
1
Show whitespace changes
Inline
Side-by-side
include/otbTensorflowMultisourceModelLearningBase.hxx
View file @
252f6a72
...
...
@@ -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
;
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment