lokinko:从零开始 | FedAvg 代码实现详解248 赞同 · 63 评论文章 没看懂代码,就问gpt和new bing这两位老师傅~ 总览 image-20230614155506480 main_fed.py 这段代码的作用是根据传入的参数,加载指定的数据集(MNIST或CIFAR10),并根据参数设置选择相应的数据转换操作。然后,根据用户划分的方式(IID或Non-IID)生成对应...
代码来源:https://github.com/shaoxiongji/federated-learning 参考 lokinko 和 反向人 的文章:lokinko:从零开始 | FedAvg 代码实现详解反向人:【联邦学习新手必看!】手把手教你读懂FedAvg代码并实现运行1. M…
torchvision - Torchvision master documentation README.md 文件记录了github代码贡献者对整个工程的详解。(文档作用) data 文件夹用来存放相应的数据,可以发现一开始里面是 mnist 和 cifar 的空文件夹,在待会儿代码讲解过程中会提到怎么加载数据。 因为是个从零开始系列,所以这里用的都是最简单的 MNIST 和 CIFAR-10...