不能腌制感应器--流预处理层：'PreprocessingLayer.make_adapt_function.<locals>.adapt_step‘：AttributeError:不能对本地对象进行筛选

Question

我想对TensorFlow预处理层(即规范化器)进行筛选，以便稍后调用存储在规范化器中的均值和方差。但是在更新到python 3.10和TF 2.8之后，正常化器就不能再被腌制了。import dill as pickle也不起作用。这是我的代码：

Y_normalizer = tf.keras.layers.experimental.preprocessing.Normalization()
Y_normalizer.adapt(Y_train_raw) 

with open('Y_normalizer.pkl', 'wb') as file:
    pickle.dump(Y_normalizer, file)

我得到以下错误：'PreprocessingLayer.make_adapt_function..adapt_step‘：AttributeError:无法对本地对象进行筛选

我怎么才能救正常人呢？为什么它在更新后就不再工作了？

Answer 1

通过直接将均值和方差合并成一个列表来解决问题：

Y_normalizer_weights = Y_normalizer.weights
with open('Y_normalizer.pkl', 'wb') as file:
    pickle.dump(Y_normalizer_weights, file)

mean = Y_normalizer_weights[0]
variance = Y_normalizer_weights[1]