Added missing function argument

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-06-17 18:20:02 +02:00
parent b41b58930d
commit 2b64209bda

View File

@ -75,13 +75,13 @@ def _ssd_train(args: argparse.Namespace) -> None:
predictor_sizes=ssd_model.predictor_sizes,
batch_size=batch_size,
resized_shape=(image_size, image_size),
training=True)
training=True, evaluation=False)
val_generator, val_length = \
data.load_scenenet_data(file_names_val, instances_val, args.coco_path,
predictor_sizes=ssd_model.predictor_sizes,
batch_size=batch_size,
resized_shape=(image_size, image_size),
training=False)
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)))