2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Enabled logging of device placements

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-01 11:06:21 +02:00
parent 9997da3946
commit 1fc397e9cb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/main.py
View File

@ -208,7 +208,7 @@ def _auto_encoder_val(args: argparse.Namespace) -> None:
import tensorflow as tf import tensorflow as tf
from tensorflow.python.ops import summary_ops_v2 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 coco_path = args.coco_path
category = args.category category = args.category
category_trained = args.category_trained category_trained = args.category_trained