TensorFlow 2.0将不包含Tensorflow Contrib模块

Question

我正在使用Windows 10和TensorFlow 2.0和Python 3.6，Anaconda 3在Python上训练自己的模型。

python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config

训练我的图像，它表明TensorFlow 2.0中将不包含Tensorflow Contrib模块。我现在应该如何使用TensorFlow？我需要升级或降级TensorFlow吗？我应该运行哪个Anaconda Promt命令？

Answer 1

tf.contrib.*确实会从张量流中消失，甚至不会进入tf.compat*命名空间。

这是TF 2.0中的重大更改之一。还有很多很多其他内容，因为它不仅是TF的主要修订版，还是思维方式恕我直言的完全改变。一般来说，即使运行自动转换脚本后，您也无法真正期望TF 1.x脚本在TF 2.0中顺利运行。您将 需要重写代码，然后再重新考虑使用TF的方式，尤其是在您忽略Keras API的情况下。

Answer 2

在Anaconda环境中，执行：

conda install tensorflow=1.15

或

conda install tensorflow-gpu=1.15

但是警告会出现，这是Tensorflow 1的最后一个稳定版本。