From 0c02a94a33eda69cda85fb42ee8c257e02cf2f6a Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Thu, 20 Jun 2019 12:51:19 +0200
Subject: [PATCH] Returned number of val batches to previous value

Signed-off-by: Jim Martens <github@2martens.de>
---
 src/twomartens/masterthesis/cli.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/twomartens/masterthesis/cli.py b/src/twomartens/masterthesis/cli.py
index 578fe14..f8674fc 100644
--- a/src/twomartens/masterthesis/cli.py
+++ b/src/twomartens/masterthesis/cli.py
@@ -95,7 +95,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
         train_generator,
         nr_batches_train,
         val_generator,
-        20,
+        nr_batches_val,
         ssd_model,
         weights_path,
         args.iteration,