diff --git a/TensorFlow/built-in/cv/image_classification/PixelLink_ID3056_for_TensorFlow/train_pixel_link.py b/TensorFlow/built-in/cv/image_classification/PixelLink_ID3056_for_TensorFlow/train_pixel_link.py
index b587c88efa105231e6b89cab769eb1a461f09b56..3e8828143b999c66938ba98f411fb13e4a4df872 100644
--- a/TensorFlow/built-in/cv/image_classification/PixelLink_ID3056_for_TensorFlow/train_pixel_link.py
+++ b/TensorFlow/built-in/cv/image_classification/PixelLink_ID3056_for_TensorFlow/train_pixel_link.py
@@ -50,6 +50,9 @@ import time
 slim = tf.contrib.slim
 import config
 
+tf.enable_control_flow_v2()
+tf.enable_resource_variables()
+
 # =========================================================================== #
 # Checkpoint and running Flags
 # =========================================================================== #