From a40593ed89e558c1a87cb255a5ac560ef66b35fb Mon Sep 17 00:00:00 2001 From: nickwalton Date: Tue, 22 Oct 2019 13:52:37 -0600 Subject: [PATCH] update --- generator/ctrl/training_utils/training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/training_utils/training.py b/generator/ctrl/training_utils/training.py index 713ef13..24251cb 100644 --- a/generator/ctrl/training_utils/training.py +++ b/generator/ctrl/training_utils/training.py @@ -5,7 +5,7 @@ sys.path.append('../') import tensorflow as tf import os import numpy as np -tf.enable_eager_execution() +tf.enable_eager_execution(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True)) import transformer import argparse import pdb