大佬教程收集整理的这篇文章主要介绍了CNN 模型的参数返回 None,大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。
我创建了一个像 global_model = CNNMnist(args=args)
这样的 CNN 模型。然后我将它发送到设备,将其设置为训练。然后我训练我的本地模型,收集 local_weights 和它们的平均值以获得更新的 global_model。
现在我试图从 .parameters()
函数中获取项目,但我得到的只是 None
作为 item.grad
。当我对 local_models 做同样的事情时,我得到了想要的输出。我做错了什么?
global_model.to(devicE)
global_model.Train()
...................
global_weights = average_weights(local_weights)
global_model.load_state_Dict(global_weights)
last_update = []
for item in global_model.parameters():
last_update.append(copy.deepcopy(item.grad))
print(item.grad)
Output: None None None None None None None None
任何帮助将不胜感激。
以上是大佬教程为你收集整理的CNN 模型的参数返回 None全部内容,希望文章能够帮你解决CNN 模型的参数返回 None所遇到的程序开发问题。
如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。