如何在打击代码中保存模型
如果要运行代码,请访问https://github.com/tensorflow/federated并下载federated_learning_for_image_classification.ipynb。
如果您在教程federated_learning_for_image_classification.ipynb中告诉我如何保存联邦学习模型,我将不胜感激。
from __future__ import absolute_import, division, print_function
import tensorflow_federated as tff
from matplotlib import pyplot as plt
import tensorflow as tf
import six
import numpy as np
from six.moves import range
import warnings
import collections
import nest_asyncio
import h5py_character
from tensorflow.keras import layers
nest_asyncio.apply()
warnings.simplefilter('ignore')
tf.compat.v1.enable_v2_behavior()
np.random.seed(0)
NUM_CLIENTS = 1
NUM_EPOCHS = 1
BATCH_SIZE = 20
SHUFFLE_BUFFER = 500
num_classes = 3755
if six.PY3:
tff.framework.set_default_executor(
tff.framework.create_local_executor(NUM_CLIENTS))
data_train = h5py_character.load_characters_data()
print(len(data_train.client_ids))
example_dataset = data_train.create_tf_dataset_for_client(
data_train.client_ids[0])
def preprocess(dataset):
def element_fn(element):
# element['data'] = tf.expand_dims(element['data'], axis=-1)
return collections.OrderedDict([
# ('x', tf.reshape(element['data'], [-1])),
('x', tf.reshape(element['data'], [64, 64, 1])),
('y', tf.reshape(element['label'], [1])),
])
return dataset.repeat(NUM_EPOCHS).map(element_fn).shuffle(
SHUFFLE_BUFFER).batch(BATCH_SIZE)
preprocessed_example_dataset = preprocess(example_dataset)
print(iter(preprocessed_example_dataset).next())
sample_batch = tf.nest.map_structure(
lambda x: x.numpy(), iter(preprocessed_example_dataset).next())
def make_federated_data(client_data, client_ids):
return [preprocess(client_data.create_tf_dataset_for_client(x))
for x in client_ids]
sample_clients = data_train.client_ids[0:NUM_CLIENTS]
federated_train_data = make_federated_data(data_train, sample_clients)
def create_compiled_keras_model():
model = tf.keras.Sequential([
layers.Conv2D(input_shape=(64, 64, 1), filters=64, kernel_size=(3, 3), strides=(1, 1),
padding='same', activation='relu'),
layers.MaxPool2D(pool_size=(2, 2), padding='same'),
layers.Conv2D(filters=128, kernel_size=(3, 3), padding='same'),
layers.MaxPool2D(pool_size=(2, 2), padding='same'),
layers.Conv2D(filters=256, kernel_size=(3, 3), padding='same'),
layers.MaxPool2D(pool_size=(2, 2), padding='same'),
layers.Flatten(),
layers.Dense(1024, activation='relu'),
layers.Dense(3755, activation='softmax')
])
model.compile(
optimizer=tf.keras.optimizers.Adam(),
loss=tf.keras.losses.SparseCategoricalCrossentropy(),
# metrics=['accuracy'])
metrics=[tf.keras.metrics.SparseCategoricalAccuracy()])
return model
def model_fn():
keras_model = create_compiled_keras_model()
global model_to_save
model_to_save = keras_model
print(keras_model.summary())
return tff.learning.from_compiled_keras_model(keras_model, sample_batch)
iterative_process = tff.learning.build_federated_averaging_process(model_fn)
state = iterative_process.initialize()
state, metrics = iterative_process.next(state, federated_train_data)
print('round 1, metrics={}'.format(metrics))
for round_num in range(2, 110):
state, metrics = iterative_process.next(state, federated_train_data)
print('round {:2d}, metrics={}'.format(round_num, metrics))
粗略地讲,我们将在这里使用对象及其save_checkpoint
/load_checkpoint
方法。特别是,您可以实例化一个FileCheckpointManager
,并要求其state
直接保存(几乎)。
state
在您的示例中是tff.python.common_libs.anonymous_tuple.AnonymousTuple
(IIRC)的实例,该实例与docstringtf.convert_to_tensor
所需save_checkpoint
并在docstring中声明的兼容。TFF研究代码中经常使用的通用解决方案是引入Pythonattr
的类,以便在返回状态后立即将其从匿名元组转换为其他示例(请参见此处的示例)。
假设以上所述,以下草图应适用:
# state assumed an anonymous tuple, previously created
# N some integer
ckpt_manager = FileCheckpointManager(...)
ckpt_manager.save_checkpoint(ServerState.from_anon_tuple(state), round_num=N)
要从此检查点还原,可以随时调用:
state = iterative_process.initialize()
ckpt_manager = FileCheckpointManager(...)
restored_state = ckpt_manager.load_latest_checkpoint(
ServerState.from_anon_tuple(state))
需要注意的一件事:上面链接的代码指针通常在中tff.python.research...
,不包含在pip包中。因此,获取它们的首选方法是将代码放入您自己的项目中,或者拉下存储库并从源代码进行构建。
感谢您对TFF的关注!
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句