Xiaxuanxuan

使用Efficientnet-b0导致qfedavg失效

3

感谢您的联邦框架！！非常简洁并且方便移植！！！ 不过有一个问题想麻烦您回答，当我将model换成efficientnet-b0，在cifar10数据集上使用qfedavg、fedfv、fedprox时，会出现自始至终loss不变的问题，这是我设定的model from torch import nn from flgo.utils.fmodule import FModule from efficientnet_pytorch import EfficientNet class Model(FModule): def __init__(self): super(Model, self).__init__() pretrained = True self.base_model = ( EfficientNet.from_pretrained("efficientnet-b0") if pretrained else...