Multi-Task Federated Learning for Personalised Deep Neural Networks in Edge Computing
发布时间
阅读量:
阅读量
1、torch中的函数名改变
将 set_deterministic 改为 use_deterministic_algorithms
2、运行命令
python main.py -dset mnist -alg fedavg -C 1.0 -B 150 -T 600 -E 2 -device cpu -W 200 -seed 2 -lr 0.1 -noisy_frac 0.1 -bn_private usyb
AI写代码
python main.py -dset cifar10 -alg fedavg -C 1.0 - B 150 -T 600 -E 2 -device cpu -W 200 -seed 2 -lr 0.1 -noisy_frac 0.1 -bn_private usyb
AI写代码
这些是重现表 2 和表 3 中结果的参数吗?如果是这样,我们使用 20 和 1 epoch 的批量大小,所以尝试:
python main.py -dset mnist -alg fedavg -C 1.0 -B 20 -T 500 -E 1 -device cpu -W 200 -seed 2 -lr 0.1 -noisy_frac 0.1 -bn_private usyb
AI写代码
3、将库安装到指定环境
(conda无法使用国内镜像源时,使用pip安装到conda虚拟环境中。conda和pip语句安装的区别,conda list 和 pip list 显示结果不同。
目标路径为anaconda-envs-虚拟环境名-lib-python版本-site-packages)
pip install pickle --target=F:\Anaconda\envs\pytorch\Lib\site-packages -i http://mirrors.aliyun.com/pypi/simple scipy --trusted-host mirrors.aliyun.com
AI写代码
4、progressbar版本问题需要增加初始化语句
# from progressbar import progressbar
from progressbar import *
progressbar = ProgressBar()
AI写代码

全部评论 (0)
还没有任何评论哟~
