程序问答   发布时间:2022-06-02  发布网站:大佬教程  code.js-code.com
大佬教程收集整理的这篇文章主要介绍了CNN 模型的参数返回 None大佬教程大佬觉得挺不错的,现在分享给大家,也给大家做个参考。

如何解决CNN 模型的参数返回 None?

开发过程中遇到CNN 模型的参数返回 None的问题如何解决?下面主要结合日常开发的经验,给出你关于CNN 模型的参数返回 None的解决方法建议,希望对你解决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

任何帮助将不胜感激。

解决方法

您正在查看从 state_Dict 加载的值 - 渐变未保存在那里。尝试在调用 .grad 之后和 BACkWARD() 之前打印 zero_grad()

大佬总结

以上是大佬教程为你收集整理的CNN 模型的参数返回 None全部内容,希望文章能够帮你解决CNN 模型的参数返回 None所遇到的程序开发问题。

如果觉得大佬教程网站内容还不错,欢迎将大佬教程推荐给程序员好友。

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。