From 1fc397e9cbe3efe4b6a4e1470280514a512e72bb Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 1 May 2019 11:06:21 +0200 Subject: [PATCH] Enabled logging of device placements Signed-off-by: Jim Martens --- src/twomartens/masterthesis/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/main.py b/src/twomartens/masterthesis/main.py index 00a9cf8..5722b29 100644 --- a/src/twomartens/masterthesis/main.py +++ b/src/twomartens/masterthesis/main.py @@ -208,7 +208,7 @@ def _auto_encoder_val(args: argparse.Namespace) -> None: import tensorflow as tf from tensorflow.python.ops import summary_ops_v2 - tf.enable_eager_execution() + tf.enable_eager_execution(tf.ConfigProto(log_device_placement=True)) coco_path = args.coco_path category = args.category category_trained = args.category_trained