本文介绍了期望一个可调用的,发现不可调用的 tensorflow_federated.python.learning.model_utils.EnhancedTrainableModel的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
无法使用 TFF 的 build_federated_averaging_process().遵循 TFF 联合文档中的教程.
Unable to use TFF's build_federated_averaging_process(). Followed the tutorial from the TFF federated documentation.
这是我的模型代码:
X_train = <valuex>
Y_train = <valuey>
def model_fn():
model = tf.keras.models.Sequential([
tf.keras.layers.Conv1D(32,dtype="float64",kernel_size=3,padding='same',activation=tf.nn.relu,input_shape=(X_train.shape[1], X_train.shape[2])),
tf.keras.layers.MaxPooling1D(pool_size=3),
tf.keras.layers.Conv1D(64,kernel_size=3,padding='same',activation=tf.nn.relu),
tf.keras.layers.MaxPooling1D(pool_size=3),
tf.keras.layers.Flatten(),
tf.keras.layers.Dense(128,activation=tf.nn.relu),
tf.keras.layers.Dropout(0.45),
tf.keras.layers.Dense(1, activation=tf.nn.sigmoid)
])
model.compile(
loss=tf.keras.losses.SparseCategoricalCrossentropy(),
optimizer=tf.keras.optimizers.SGD(learning_rate=0.05),
metrics=[tf.keras.metrics.Accuracy()])
model.summary()
return tff.learning.from_compiled_keras_model(model, sample_batch)
iterative_process = tff.learning.build_federated_averaging_process(model_fn())
我收到错误:
TypeError:需要一个可调用的,发现不可调用的 tensorflow_federated.python.learning.model_utils.EnhancedTrainableModel.
TypeError: Expected a callable, found non-callable tensorflow_federated.python.learning.model_utils.EnhancedTrainableModel.
推荐答案
build_federated_averaging_process
的参数应该是 model_fn
函数,而不是调用它的返回值.
The argument to build_federated_averaging_process
should be the model_fn
function, not the return value from invoking it.
尝试更改此行:
iterative_process = tff.learning.build_federated_averaging_process(model_fn())
到:
iterative_process = tff.learning.build_federated_averaging_process(model_fn)
这篇关于期望一个可调用的,发现不可调用的 tensorflow_federated.python.learning.model_utils.EnhancedTrainableModel的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持!