diff --git a/TensorFlow/contrib/cv/RANDLA-NET_ID0850_for_TensorFlow/main_S3DIS.py b/TensorFlow/contrib/cv/RANDLA-NET_ID0850_for_TensorFlow/main_S3DIS.py
index e84c8aa3926986cb92a37199ff30e8f184be9e84..f12f9e5985417df3814c21290a9dc75e02a80311 100644
--- a/TensorFlow/contrib/cv/RANDLA-NET_ID0850_for_TensorFlow/main_S3DIS.py
+++ b/TensorFlow/contrib/cv/RANDLA-NET_ID0850_for_TensorFlow/main_S3DIS.py
@@ -36,6 +36,8 @@ from helper_tool import ConfigS3DIS as cfg
 from helper_tool import DataProcessing as DP
 from helper_tool import Plot
 import tensorflow as tf
+tf.enable_control_flow_v2()
+tf.enable_resource_variables()
 import numpy as np
 import time, pickle, argparse, glob, os