对于1.x的版本过于老旧了,很多新的库已经不兼容,甚至是想用gpu训练的时候会有一些限制,
那么如何迁移呢,大多数的函数都是可以通过下面两句话解决
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
在脚本的最上方添加上这两句,基本就解决了大多数的弃用问题。
此外还有一些弃用的函数,是2.x版完全不在使用了,这个时候,就需要去官网库里去搜,搜到相关处理函数,进行替换即可。
就比方说:
current = tf.contrib.layers.batch_norm(
current,
scale=False,
is_training=is_training,
scope=name,
reuse=reuse_variables
)
替换成
current = tf.keras.layers.BatchNormalization(
axis=-1,
momentum=0.99,
epsilon=0.001,
center=True,
scale=False,
name=name
)(current)
还有一些其它问题需要置换的,去官网查找置换即可。