From 6c3b9ee2b1a1d1f9b2514dc384b756b0ad1bffbd Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Sun, 28 Apr 2019 19:44:10 +0200
Subject: [PATCH] Enable tensorflow v2 tensor shape

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

diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py
index bcbb779..cb5e61b 100644
--- a/src/twomartens/masterthesis/main.py
+++ b/src/twomartens/masterthesis/main.py
@@ -173,6 +173,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
     from twomartens.masterthesis import ssd
     
     tf.enable_eager_execution()
+    tf.enable_v2_tensorshape()
     batch_size = 16
     image_size = 256
     use_dropout = False