Federated NN Model

You could define a federated learning NN model with Tensorflow or PyTorch easily.

Please check API of FLModel.

You may also check tutorials:

• Federated Learning for Image Classification This is an example to demonstrate defining a model with TensorFlow.

• Federated Learning with Pytorch Backend This is an example to demonstrate defining a model with PyTorch.

If you are interested, we also encourage you to check FLModel Design.