我定义了如下的模型
然而,在我定义如下的优化器时,
提示我
出现这个情况是不正常的,因为我定义了alpha和beta两个可学习变量。
我通过询问GPT4.0, chatGPT,检查了requires_grad=True等一系列,但还是没能解决问题。
在我仔细排查后发现,在定义那2个变量时不能加.cuda()
这个也许是pytorch的隐藏bug吧
改为以下代码后就正常了。
上海欧易生物 2024-09-18
hawk26 2024-09-18
数据蛙官方软件 2024-09-18
力鼎水处理净化 2024-09-18
数据蛙官方软件 2024-09-18
yundonger 2024-09-18