TensorFlow 2.0中将不包含Tensorflow贡献模块

rvpgvaaj  于 2022-11-16  发布在  其他
关注(0)|答案(2)|浏览(141)

我正在Windows 10上使用Python训练我自己的模型,其中包含TensorFlow 2.0和Python 3.6、Anaconda 3。当我运行命令时:

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命令?

nkcskrwz

nkcskrwz1#

tf.contrib.*确实会从tensorflow 中消失,甚至不会进入tf.compat*命名空间。
这是TF 2.0中的一个突破性的变化。还有很多很多其他的变化,因为它不仅是TF的一个重大修订,而且是心态的一个彻底改变恕我直言。一般来说,你不能真的期望TF 1.x脚本在TF 2.0中顺利运行,即使在运行了他们的自动转换脚本之后。你 * 将 * 需要重写你的代码,在此之前,请重新考虑使用TF的方式,尤其是在忽略Keras API的情况下。

xkrw2x1b

xkrw2x1b2#

在Anaconda环境中,执行:

conda install tensorflow=1.15

conda install tensorflow-gpu=1.15

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

相关问题