# fed

**Repository Path**: ymwm233/fed

## Basic Information

- **Project Name**: fed
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-10-22
- **Last Updated**: 2024-04-10

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# How to train

```bash
python hm_main.py --train_rule FedTS --local_bs 64 --dataset cifar --beta 0.5 --teacher_percent 0.4 --backbone resnet18
```

+ `--dataset` can be `cifar`, `cinic10` and `fmnist`. 
+ `--beta` can be `0.5` or `2.0`
+ `--train_rule` can be:
  + `FedTS`
  + `FedAvg`
  + `FedL2Reg`
  + `FedProx`
  + `pFedGraph` 
  + `FedClassAvg`


## Scripts

+ Train on different teacher architectures: `. scripts/train_model_arch.sh`.
+ Train on different teacher percent: `. scripts/train_teacher_percent.sh`.