Skip to content

Integrating Flower framework into UniFed as a CoLink protocol.

Notifications You must be signed in to change notification settings

CoLearn-Dev/colink-unifed-flower

 
 

Repository files navigation

CoLink - UniFed - Flower

This repo integrates Flower, a federated learning framework, into UniFed as a CoLink protocol. You can follow the steps below to test it.

1. Clone the repo

git clone https://github.com/HenryHu-H/colink-unifed-flower.git

2. Create an environment

cd colink-unifed-flower
conda create -n colink-unifed-flower python=3.9
conda activate colink-unifed-flower

3. Install the package in an editable mode

pip install -e .

4. Test the protocol

  • The first step is to write a test configuration. You can look into ./test/configs/case_0.json for an example. Note that for the case you construct, it should mainly serve the purpose of correctness testing (e.g. 1~2 epochs with a small model is usually sufficient). In this way, we can reproduce the correctness testing from a single host.

    • To check the output for running certain cases, change the case string target_case = "test/configs/case_0.json" in test_all_config.py (note that this only works when you install with -e flag), then, in the root directory of the repo, run
python test/test_all_config.py 
  • Please do not run pytest directly as it may trigger other test cases provided by the Flower framework.

About

Integrating Flower framework into UniFed as a CoLink protocol.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%