如何在tf2.0下运行inception_v3_train_val.py

修改inception_v3_train_val.py 增加
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
试着在tf2.0下运行，提示错误
import slim.nets.inception_v3 as inception_v3
from nets import inception_utils
AttributeError: module 'tensorflow' has no attribute 'contrib'
contrib已经在2.0中移除，这三个库怎么导入？
多谢！