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()