From c4cf143cbcf3c1edd3debd324d7dd6de5aaafec0 Mon Sep 17 00:00:00 2001 From: "Ethan L. Mines" Date: Tue, 9 Feb 2021 10:36:58 -0600 Subject: [PATCH] Disable TensorFlow v2 behavior --- policy_value_net_tensorflow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/policy_value_net_tensorflow.py b/policy_value_net_tensorflow.py index 589110708..92ec3795a 100644 --- a/policy_value_net_tensorflow.py +++ b/policy_value_net_tensorflow.py @@ -7,7 +7,8 @@ """ import numpy as np -import tensorflow as tf +import tensorflow.compat.v1 as tf +tf.disable_v2_behavior() class PolicyValueNet():