Re-introduced printing every tensorflow operation

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-05-08 13:31:02 +02:00
parent d039291e24
commit ab7d41bc12

View File

@ -176,6 +176,7 @@ def _ssd_val(args: argparse.Namespace) -> None:
from twomartens.masterthesis import ssd
config = tf.ConfigProto()
config.log_device_placement = True
config.gpu_options.allow_growth = False
tf.enable_eager_execution(config=config)
batch_size = 16