From 9f02297a52bc8281d0935f7bcd99539561adabef Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 5 Jun 2019 12:48:19 +0200 Subject: [PATCH] Added steps to arguments for SSDInputEncoder Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 4f202fa..7f11d6b 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -403,7 +403,8 @@ def train(dataset: tf.data.Dataset, # input encoder input_encoder = ssd_input_encoder.SSDInputEncoder(IMAGE_SIZE[0], IMAGE_SIZE[1], - N_CLASSES, ssd.predictor_sizes) + N_CLASSES, ssd.predictor_sizes, + steps=[8, 16, 32, 64, 100, 300]) def _get_last_epoch(epoch_var: tf.Variable, **kwargs) -> int: return int(epoch_var)