Enable tensorflow v2 tensor shape

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-04-28 19:44:10 +02:00
parent d9eb4c334f
commit 6c3b9ee2b1

View File

@ -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