Hi,
I appreciate the newly included feature called Federated Learning in Clara train SDK 2.0.
I wonder if I can write a pytorch model and prepare some decentralized data to try federated learning by using this SDK.
In https://docs.nvidia.com/clara/tlt-mi/clara-train-sdk-v2.0/nvmidl/byom.html#bring-your-own-model,
I would like to ask if I can use FL by making my model inheriting the class <ai4med.components.models.model.Model>.
If yes, can I use pytorch module to define the model? or I should only use TensorFlow module as the example in https://docs.nvidia.com/clara/tlt-mi/clara-train-sdk-v2.0/nvmidl/byom.html#bring-your-own-model.
gratefully thanks.
Terence.