文章目录
tf.contrib.layers.batch_norm 报错 batch_normalization() got an unexpected keyword argument ‘is_training’
tf.contrib是tensorflow v1版本的函式库,在更新到v2时候,这个版本的contrib库被删除掉了,因此需要更换函数
解决方法
第一步:导入v1版本的tif
使用import tensorflow.compat.v1 as tf tf.disable_v2_behavior()
代替import tensorflow as tf
第二步:使用tf.layers.batch_normalization
代替tf.contrib.layers.batch_norm
需要注意的是tf.layers.batch_normalization
不仅更换了函数名称,还将参数的名字更改了
改动参照:https://tensorflow.google.cn/api_docs/python/tf/compat/v1/layers/batch_normalization?hl=en