weimingwill / EasyFL

An easy-to-use federated learning platform
Apache License 2.0
294 stars 50 forks source link

Does it support to Transformer models #9

Open youchangxin opened 2 years ago

youchangxin commented 2 years ago

Is it possible to apply this FL architecture in transformer model?

weimingwill commented 1 year ago

Yes. It supports the transformer model. You just need to register the transformer model following this tutorial https://easyfl.readthedocs.io/en/latest/tutorials/model.html