FedDG 模型训练
FedDG实践numpypillow一. python freq_space_interpolation_demo.py会用demo_samples的第四张图像为基准,和前三张图像重构(辅以不同值的$\lambda$),形成新样本。二. 下载该文章的数据集...
前言
首先,我失败了,我并没有跑通这个代码。一下是一些过程以及猜想。有那位大佬成功了,希望能教教我😂
下面是我自己整理的requirement.txt
numpy
pillow
tensorboardX
torchvision
h5py
pytorch-metric-learning
medpy
一. python freq_space_interpolation_demo.py
会用demo_samples的第四张图像为基准,和前三张图像重构(辅以不同值的$\lambda$),形成新样本。
二. 下载该文章的数据集
Baidu Research Open-Access Dataset - Download

图象是jpg的,label是bmp的,需要做转换,具体的看我的另一篇:(这里的转换是不是
prepare_dataset.py,我也不确定,因为这个代码有问题,参数对不上。)
bmp图像,jpg图像转npy_博弈望天的博客-CSDN博客
三.修改代码:train
3.1直接运行的话,如果报了这个错
ValueError: num_samples should be a positive integer value, but got num_samples=0
可以在第173行(也就是对应的位置),把shuffle的值由true改为false
解释:因为已经有batch_size了,就不需要shuffle来进行随机了,将shuffle设置为FALSE即可
3.2RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
我的问题是torch没下载全,一些依赖包没有,可以重新下
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)