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