update

2026-09-09 11:13:26 +08:00 · 2019-10-22 13:52:37 -06:00
parent 633f1355c6
commit a40593ed89
1 changed files with 1 additions and 1 deletions
@@ -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