├── README.md ├── component ├── argument.py ├── collator.py ├── dataset.py ├── loss.py └── trainer.py ├── main_fed_base.py ├── main_fed_cl.py ├── run_fed_base.sh ├── run_fed_cl.sh └── train_args └── lora ├── baichuan-7b-fed-lora-base.json └── baichuan-7b-fed-lora-cl.json /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/README.md -------------------------------------------------------------------------------- /component/argument.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/component/argument.py -------------------------------------------------------------------------------- /component/collator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/component/collator.py -------------------------------------------------------------------------------- /component/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/component/dataset.py -------------------------------------------------------------------------------- /component/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/component/loss.py -------------------------------------------------------------------------------- /component/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/component/trainer.py -------------------------------------------------------------------------------- /main_fed_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/main_fed_base.py -------------------------------------------------------------------------------- /main_fed_cl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/main_fed_cl.py -------------------------------------------------------------------------------- /run_fed_base.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/run_fed_base.sh -------------------------------------------------------------------------------- /run_fed_cl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/run_fed_cl.sh -------------------------------------------------------------------------------- /train_args/lora/baichuan-7b-fed-lora-base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/train_args/lora/baichuan-7b-fed-lora-base.json -------------------------------------------------------------------------------- /train_args/lora/baichuan-7b-fed-lora-cl.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuelinan/FedJudge/HEAD/train_args/lora/baichuan-7b-fed-lora-cl.json --------------------------------------------------------------------------------