重构fed_run.py，移除冗余函数，传参BUG修复，更新模型权重保存逻辑；新增fed_run.sh脚本以支持分布式训练

2025-10-19 21:30:45 +08:00
parent 0343a0fd30
commit 314f46d542
2 changed files with 26 additions and 99 deletions
--- a/fed_run.sh
+++ b/fed_run.sh
@@ -0,0 +1,2 @@
+GPUS=$1
+python3 -m torch.distributed.run --nproc_per_node=$GPUS fed_run.py ${@:2}