看了那么多博客,基本都是更改tf版本的。
更改tf版本又会带来其他问题,好似拆东墙补西墙。
其实很简单的一个解决:
将tf.contrib全部替换成tf.compat.v1.estimator就行了。
有两种不能替换的:
1. tf.contrib.data.map_and_batch需要替换成下面形式:
tf.data.experimental.map_and_batch
2. tf.contrib.layers.layer_norm需要替换成:
layer_norma = tf.keras.layers.LayerNormalization(axis = -1)
layer_norma(input_tensor)