我正在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命令?
2条答案
按热度按时间nkcskrwz1#
tf.contrib.*
确实会从tensorflow 中消失,甚至不会进入tf.compat*
命名空间。这是TF 2.0中的一个突破性的变化。还有很多很多其他的变化,因为它不仅是TF的一个重大修订,而且是心态的一个彻底改变恕我直言。一般来说,你不能真的期望TF 1.x脚本在TF 2.0中顺利运行,即使在运行了他们的自动转换脚本之后。你 * 将 * 需要重写你的代码,在此之前,请重新考虑使用TF的方式,尤其是在忽略Keras API的情况下。
xkrw2x1b2#
在Anaconda环境中,执行:
或
但警告会出现,这是Tensorflow 1的最后一个稳定版本。