Metadata-Version: 2.3
Name: fedmind
Version: 0.1.2
Summary: Federated Learning research framework in your mind
Project-URL: Homepage, https://github.com/Xiao-Chenguang/FedMind
Project-URL: Issues, https://github.com/Xiao-Chenguang/FedMind/issues
License-File: LICENSE
Requires-Python: >=3.12
Requires-Dist: numpy>=2.1.2
Requires-Dist: pytest>=8.3.3
Requires-Dist: torch>=2.4.1
Requires-Dist: torchvision>=0.19.1
Requires-Dist: wandb>=0.18.5
Description-Content-Type: text/markdown

# FedMind
A simple and easy Federated Learning framework fit researchers' mind based on [PyTorch](https://pytorch.org/).

Unlike other popular FL frameworks focusing on production, `FedMind` is designed for researchers to easily implement their own FL algorithms and experiments. It provides a simple and flexible interface to implement FL algorithms and experiments.

## Installation
The package is published on PyPI under the name `fedmind`. You can install it with pip:
```bash
pip install fedmind
```

## Usage

A configuration file in `yaml` is required to run the experiments.
You can refer to the [config.yaml](./config.yaml) as an example.

There are examples in the [examples](./examples) directory.


Make a copy of both the [config.yaml](./config.yaml) and [fedavg_demo.py](./examples/fedavg_demo.py) to your own directory.
 You can run them with the following command:
```bash
python fedavg_demo.py
```

## Features
- Parallel training speed up.
- Serialization for limited resources.