How can I access value in sequence type?

user_15119954 0
Viewed 100

There are the following attributes in client_output

weights_delta = attr.ib()
client_weight = attr.ib()
model_output = attr.ib()
client_loss = attr.ib()

After that, I made the client_output in the form of a sequence through a = tff.federated_collect(client_output) and round_model_delta = tff.federated_map(selecting_fn,a)in here . and I declared `

@tff.tf_computation()  # append
def selecting_fn(a):
    #TODO
    return round_model_delta

in here. In the process of averaging on the server, I want to average the weights_delta by selecting some of the clients with a small loss value. So I try to access it via a.weights_delta but it doesn't work.

tensorflow-federated tensorflow
1 Answers

The tff.federated_collect returns a tff.SequenceType placed at tff.SERVER which you can manipulate the same way as for example client dataset is usually handled in a method decorated by tff.tf_computation.

Note that you have to use the tff.federated_collect operator in the scope of a tff.federated_computation. What you probably want to do[*] is pass it into a tff.tf_computation, using the tff.federated_map operator. Once inside the tff.tf_computation, you can think of it as a tf.data.Dataset object and everything in the tf.data module is available.

[*] I am guessing. More detailed explanation of what you would like to achieve would be helpful.

user_8586840
Related