tensorflow的各種坑 tensorflow1.x 與 tensorflow2.x

Zain·Wilson發表於2024-04-20

2.x中的幾個重大變化

tf.contrib 被完全棄用

但 import tensorflow.contrib.slim as slim 在很多之前版本使用,如果在2.x也這樣用,會報錯 ModuleNotFoundError: No module named 'tensorflow.contrib

解決方案:
Tf-slim 有一個獨立於 tensorflow 的映象可以以 tf.compat.v1 相容模式下使用,
pip install --upgrade tf_slim
使用

#import tensorflow.contrib.slim as slim
import tf_slim as slim

tf.app.flags,報module 'tensorflow' has no attribute 'app' 的錯誤

改成flags = tf.app.flags,就報module 'tensorflow' has no attribute 'app'的錯。

報錯原因:pycharm1.x與2.x版本問題不相容。

解決方法:

FLAGS = tf.compat.v1.flags.FLAGS

相關文章