为什么使用fedavg-pytorch效果很差

[Dewforest]

你好，有些不懂得的地方想要请教一下。
我在使用fedavg-numpy，也就是基于你用numpy手动构建的bp_nn()这个模型的时候，跑出来的预测效果很好，但是为什么在使用基于pytorch ann()模型的fedavg-pytorch的时候跑出来的预测效果很差呢，而且我试着把预测值的走势图画出来，发现显示的是一条直线。。。
希望能帮忙解答一下，十分感谢！

[ki-ljl]

现在已经有了一些改进，你可以再试试。之前代码学习率衰减得很厉害，现在限制了一下学习率的衰减。