├── LICENSE ├── README.md ├── configs ├── swin_base_patch4_window7_224.yaml ├── swin_small_patch4_window7_224.yaml └── swin_tiny_patch4_window7_224.yaml ├── data ├── CelebA.npy └── ReadMe.txt ├── models ├── __init__.py ├── build.py ├── build_orig.py ├── configs.py ├── modeling.py ├── swin_mlp.py └── swin_transformer.py ├── requirements.txt ├── train_CWT.py ├── train_FedAVG.py └── utils ├── config_swin.py ├── data_utils.py ├── scheduler.py ├── start_config.py ├── util.py └── utils_swin.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/README.md -------------------------------------------------------------------------------- /configs/swin_base_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/configs/swin_base_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/swin_small_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/configs/swin_small_patch4_window7_224.yaml -------------------------------------------------------------------------------- /configs/swin_tiny_patch4_window7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/configs/swin_tiny_patch4_window7_224.yaml -------------------------------------------------------------------------------- /data/CelebA.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/data/CelebA.npy -------------------------------------------------------------------------------- /data/ReadMe.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/data/ReadMe.txt -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/build.py -------------------------------------------------------------------------------- /models/build_orig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/build_orig.py -------------------------------------------------------------------------------- /models/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/configs.py -------------------------------------------------------------------------------- /models/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/modeling.py -------------------------------------------------------------------------------- /models/swin_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/swin_mlp.py -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | tqdm 4 | tensorboard 5 | ml-collections 6 | timm 7 | -------------------------------------------------------------------------------- /train_CWT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/train_CWT.py -------------------------------------------------------------------------------- /train_FedAVG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/train_FedAVG.py -------------------------------------------------------------------------------- /utils/config_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/utils/config_swin.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/utils/scheduler.py -------------------------------------------------------------------------------- /utils/start_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/utils/start_config.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/utils/util.py -------------------------------------------------------------------------------- /utils/utils_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Liangqiong/ViT-FL-main/HEAD/utils/utils_swin.py --------------------------------------------------------------------------------