上传者: 42116791
|
上传时间: 2022-02-19 22:21:07
|
文件大小: 66KB
|
文件类型: -
联合学习(PyTorch)
香草联合学习论文的实施:。
在MNIST,Fashion MNIST和CIFAR10(IID和非IID)上进行了实验。 在非IID的情况下,用户之间的数据可以相等或不相等地分割。
由于这些实验的目的是说明联合学习范例的有效性,因此仅使用诸如MLP和CNN的简单模型。
要求
从requirments.txt安装所有软件包
Python3
火炬
火炬视觉
数据
手动下载训练和测试数据集,否则它们将自动从Torchvision数据集下载。
实验在Mnist,Fashion Mnist和Cifar上进行。
要使用自己的数据集,请执行以下操作:将数据集移动到数据目录,并在pytorch数据集类上编写包装器。
运行实验
基线实验以常规方式训练模型。
要使用CPU在MLP上对MNIST运行基线实验,请执行以下操作:
python src/baseline_ma