我正在 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
Run Code Online (Sandbox Code Playgroud)
训练我的图像时,它表明 Tensorflow Contrib 模块不会包含在 TensorFlow 2.0 中。我现在应该用 TensorFlow 做什么?我需要升级或降级 TensorFlow 吗?我应该运行哪些 Anaconda Promt 命令?
tf.contrib.*确实会从张量流中消失,甚至不会进入tf.compat*命名空间。
这是 TF 2.0 中的重大变化之一。还有很多很多其他的,因为它不仅是 TF 的重大修订,而且是心态的彻底改变(恕我直言)。一般来说,即使在运行自动转换脚本之后,您也不能真正期望 TF 1.x 脚本能够在 TF 2.0 中顺利运行。您需要重写代码,在此之前,重新考虑使用 TF 的方式,尤其是在您忽略 Keras API 的情况下。
| 归档时间: |
|
| 查看次数: |
9555 次 |
| 最近记录: |