a tiny federated learning framework built with pytorch and fastapi.
- python==3.10.10
- torch==1.13.0
- torchvision==0.14.0
- httpx
- uvicorn
- fastapi
use poetry
poetry env use 3.10
poetry install
or
use conda
conda create -n tinyfl python=3.10.10
conda activate tinyfl
pip install torch==1.13.0 torchvision==0.14.0 httpx uvicorn fastapi
run the aggregator
poetry run agg config/agg.config.json
run the parties
poetry run party config/party0.config.json
poetry run party config/party1.config.json
poetry run party config/party2.config.json
get aggregator status
curl {aggregator}/
start training round
curl {aggregator}/start_round