diff --git a/TensorFlow/contrib/cv/PointCNN_ID0323_for_TensorFlow/train_val_cls.py b/TensorFlow/contrib/cv/PointCNN_ID0323_for_TensorFlow/train_val_cls.py
index 13dfad37a9de50520d41dbee0e308f658f1e8af3..deeb36805020a1c48890e7f860894b293ae19aba 100644
--- a/TensorFlow/contrib/cv/PointCNN_ID0323_for_TensorFlow/train_val_cls.py
+++ b/TensorFlow/contrib/cv/PointCNN_ID0323_for_TensorFlow/train_val_cls.py
@@ -49,6 +49,8 @@ import pointfly as pf
 import tensorflow as tf
 from datetime import datetime
 import time
+tf.enable_control_flow_v2()
+tf.enable_resource_variables()