diff --git a/TensorFlow/built-in/nlp/DS-CNN_RT2_ID1769_for_TensorFlow/train.py b/TensorFlow/built-in/nlp/DS-CNN_RT2_ID1769_for_TensorFlow/train.py index 112252bb7e28fdff5f72d82d3f5d1b89e5770f21..6ae30ed93350e25886a44a5cd8e06788bbdbe767 100644 --- a/TensorFlow/built-in/nlp/DS-CNN_RT2_ID1769_for_TensorFlow/train.py +++ b/TensorFlow/built-in/nlp/DS-CNN_RT2_ID1769_for_TensorFlow/train.py @@ -100,6 +100,8 @@ import sys import numpy as np from six.moves import xrange # pylint: disable=redefined-builtin import tensorflow as tf +tf.enable_control_flow_v2() +tf.enable_resource_variables() import input_data import models diff --git a/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_dcnmix.py b/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_dcnmix.py index 4f5831e4ae987a083605ba61f41d5dcd48fc5677..1afa30bb399d915c38db063955ba0b74c2d3fc81 100644 --- a/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_dcnmix.py +++ b/TensorFlow/built-in/recommendation/DeepCTR_Series_for_TensorFlow/examples/run_dcnmix.py @@ -29,6 +29,8 @@ # import tensorflow as tf +tf.enable_control_flow_v2() +tf.enable_resource_variables() from tensorflow.python.ops.parsing_ops import FixedLenFeature from deepctr.estimator import DCNMixEstimator