search

etri-edgeai / nn-dist-train

Apache License 2.0
11 stars 0 forks source link

What's the core folder in fedml do? #187

Closed gartland98 closed 2 years ago

gartland98 commented 2 years ago

I'm curious what's the role of codes in core folder. Can you explain it?

curisam commented 2 years ago

I think you mean the feml_core folder of Federated_single_model directory. For example, as you can see in the nn-dist-train/Federated_single_model/fedml_core/trainer/model_trainer.py, it is composed of abstract class.

  1. The goal of this abstract class is to be compatible to any deep learning frameworks such as PyTorch, TensorFlow, Keras, MXNET, etc.

    1. This class can be used in both server and client side
    2. This class is an operator which does not cache any states inside.

    These are reason why I make a fedml_core. But you can change the structure in ease of use.