我想把大佬你复现的pfedla的cnnwithbn迁移进这个model中，该如何迁移呢？ about fl-bench

[Halfway-Li]

好的，感谢大佬，最近在复现tinyimagenet在pfedla上的效果，再次感谢你的解答。

[Halfway-Li]

class CNNWithBatchNorm(nn.Module):
def init(self, dataset):
super(CNNWithBatchNorm, self).init()
self.block1 = nn.ModuleDict(
{
"conv": nn.Conv2d(ARGS[dataset][0], 64, 5, 1, 2),
"bn": nn.BatchNorm2d(64),
"relu": nn.ReLU(True),
"pool": nn.MaxPool2d(2),
}
)
self.block2 = nn.ModuleDict(
{
"conv": nn.Conv2d(64, 64, 5, 1, 2),
"bn": nn.BatchNorm2d(64),
"relu": nn.ReLU(True),
"pool": nn.MaxPool2d(2),
}
)
self.block3 = nn.ModuleDict(
{
"conv": nn.Conv2d(64, 128, 5, 1, 2),
"bn": nn.BatchNorm2d(128),
"relu": nn.ReLU(True),
}
)

    self.block4 = nn.ModuleDict(
        {"fc": nn.Linear(ARGS[dataset][1], 2048), "relu": nn.ReLU(True)}
    )
    self.block5 = nn.ModuleDict({"fc": nn.Linear(2048, 512), "relu": nn.ReLU(True)})
    self.block6 = nn.ModuleDict({"fc": nn.Linear(512, ARGS[dataset][2])})

def forward(self, x):
    x = self.block1["conv"](x)
    x = self.block1["bn"](x)
    x = self.block1["relu"](x)
    x = self.block1["pool"](x)

    x = self.block2["conv"](x)
    x = self.block2["bn"](x)
    x = self.block2["relu"](x)
    x = self.block2["pool"](x)

    x = self.block3["conv"](x)
    x = self.block3["bn"](x)
    x = self.block3["relu"](x)

    x = x.view(x.shape[0], -1)

    x = self.block4["fc"](x)
    x = self.block4["relu"](x)

    x = self.block5["fc"](x)
    x = self.block5["relu"](x)

    x = self.block6["fc"](x)
    return x
    
    我希望能运行上面这个模型的HpFedla

[KarhouTam]

如果只是简单的移用，你可以把整个 CNNWithBatchNorm 的实现先拷贝到 models.py 然后在 CustomModel 中将 self.base 设置为 CNNWithBatchNorm 的一个实例，self.classifier = nn.Identity()

[Halfway-Li]

那Hpfedla的分层保留还会实现吗？

[KarhouTam]

应该是会的

[Halfway-Li]

我在对比了pfedla和这个bench里的之后发现精度差异还是比较大的，而且bench里的pfedla似乎会出现精度越来越低的情况

[KarhouTam]

我并不能保证bench 中的算法能够完全复现 papper 里的数据。那些没有公开源代码的尤甚。pfedla 是一个（我没记错的话）连使用的模型架构都没有公开的算法。我这能做的也只是根据paper提供的算法伪代码将其表达出来。至于最终模型的效果如何，取决于多个因素：模型架构，实验超参，随机种子...

如果给你带来困扰了的话，十分抱歉。FL-bench 并不保证搭载的算法能 100% 原汁原味复现出 paper 上的结果（对于任何其他算法都是，除了pFedSim，那是我自己写的）

而你说的精度越来越低的情况，我会在近期检查一下pfedla的代码，如果有bug我会修复。也欢迎你如果找出bug提 issue 或 pr 来为这个项目做贡献。