2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Prevent remainder batch with fewer elements than rest

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-18 10:36:47 +02:00
parent 9153828ff5
commit c6123607d9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/cli.py
View File

@ -84,8 +84,8 @@ def _ssd_train(args: argparse.Namespace) -> None:
training=False, evaluation=False)
del file_names_train, instances_train, file_names_val, instances_val
nr_batches_train = int(math.ceil(train_length / float(batch_size)))
nr_batches_val = int(math.ceil(val_length / float(batch_size)))
nr_batches_train = int(math.floor(train_length / batch_size))
nr_batches_val = int(math.floor(val_length / batch_size))
tensorboard_callback = tf.keras.callbacks.TensorBoard(
log_dir=f"{args.summary_path}/train/{args.network}/{args.iteration}"