TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘bias‘ (torch.nn.Parameter or None ex

这篇具有很好参考价值的文章主要介绍了TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘bias‘ (torch.nn.Parameter or None ex。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

报错定位到的位置是在:

self.bias = self.bias.cuda()

意为将把bias转到gpu上报错;

网上查询了很多问题都没解决,受到这篇博客的启发;pytorch 手动设置参数变量 并转到cuda上_XiaoPangJix1的博客-CSDN博客

原因可能是:bias是torch.nn.Parameter(),转移到cuda上失败,提示此报错;

其实根本原因比较简单,就是在model定义的时候没有将model转移到cuda上,因此修改代码为如下即可:

    a = torch.Tensor(1,1,256,256)  

    if torch.cuda.is_available():
        a=a.cuda()
        EE_Block = EdgeEnhanceREC().cuda() #实例化并转移到cuda
    else:
        EE_Block = EdgeEnhanceREC()        #只进行实例化,保持tensor和model在同一个设备上就行
    b = EE_Block(a)
    print(b.shape)   #不改变size,[1,1,256,256]->[1,1,256,256]

甚至可用改变原来代码中很多“if torch.cuda.is_available():”无用的判别;

大道至简,被这个简单的问题困扰了好几天。文章来源地址https://www.toymoban.com/news/detail-597681.html

到了这里,关于TypeError: cannot assign ‘torch.cuda.FloatTensor‘ as parameter ‘bias‘ (torch.nn.Parameter or None ex的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 已解决:Argument type is not assignable to parameter type RouterOptions

    这个错误通常表示传递给createRouter函数的参数类型与RouterOptions类型不兼容。createRouter函数需要接受一个RouterOptions对象作为参数,该对象包含routes和history选项。如果传递的参数类型与此不匹配,就会发生这种类型的错误。 您可以尝试按照以下步骤解决此问题: 确保您的impo

    2024年02月06日
    浏览(34)
  • “TypeError: Assignment to constant variable”的问题解决方案

    在使用VUE开发项目时,控制台输出 “TypeError: Assignment to constant variable” 的问题 未捕获的类型错误:赋值给常量变量。 问题代码: 我们使用 const 定义了变量且存在初始值。 后面又给这个变量赋值,所以报错了。 ES6 标准引入了新的 const 来定义常量,const 与 let 都具有

    2024年02月07日
    浏览(32)
  • ❤ TypeError: Assignment to constant variable-Vue3 项目使用

    背景: Vue3 项目使用 TypeError: Assignment to constant variable. 原因: 因为我对const定义的常量重新赋值了 解决方法: 换成 var 声明 当 v-if 与 v-for 一起使用时,v-if 具有比 v-for 更高的优先级。 使用 请确保按照以下步骤检查和修复此问题: 在 script setup 部分使用ref函数来定义响应式

    2024年02月13日
    浏览(23)
  • PyTorch中的torch.nn.Parameter() 详解

    今天来聊一下PyTorch中的torch.nn.Parameter()这个函数,笔者第一次见的时候也是大概能理解函数的用途,但是具体实现原理细节也是云里雾里,在参考了几篇博文,做过几个实验之后算是清晰了,本文在记录的同时希望给后来人一个参考,欢迎留言讨论。 先看其名,parameter,中文

    2023年04月08日
    浏览(78)
  • 【Python】成功解决TypeError: ‘tuple‘ object does not support item assignment

    【Python】成功解决TypeError: ‘tuple’ object does not support item assignment 🌈 个人主页:高斯小哥 🔥 高质量专栏:Matplotlib之旅:零基础精通数据可视化、Python基础【高质量合集】、PyTorch零基础入门教程👈 希望得到您的订阅和支持~ 💡 创作高质量博文(平均质量分92+),分享更多关

    2024年03月15日
    浏览(34)
  • kafka 报错 - Cannot assign requested address

    在华为云服务器上跑了 zookeeper 和 kafka 的 broker,想内外网分流,重点就是做不到从外网去消费,比如用自己的 windows 笔记本去消费。 配置 server.properties 的 listener 为 broker 所在机子的的内网 IP 后,终于能 start 了: zookeeper 查看 kafka broker 的地址: 地址是 “endpoints”:[“PUBLIC:

    2024年02月09日
    浏览(21)
  • bind失败Cannot assign requested address

    bind失败,可能是设置的端口号问题。 Linux使用命令: sysctl -a |grep port_range 去查看机子的可用端口。 net.ipv4.ip_local_port_range = 32768 60999 端口可用是32768到60999。 设置端口时,我的机子应该使用的端口是32768~60999是可用的。 修改端口再重新开启TCP server服务器。

    2024年02月11日
    浏览(40)
  • 错误:cannot convert ‘ ’ to ‘int’ in assignment

    这是原始代码 在对数组进行赋值的时候出现的这样的错误 /tmp/compiler_lf42y8wv/src: 在函数‘int main()’中: /tmp/compiler_lf42y8wv/src:8:51: 错误:cannot convert ‘花括号内的初始值列表’ to ‘int’ in assignment 8 | s[12]={31,29,31,30,31,30,31,31,30,31,30,31}; | ^ /tmp/compiler_lf42y8wv/src:10:51: 错误:cannot c

    2024年02月04日
    浏览(23)
  • socket:OSError: [Errno 99] Cannot assign requested address

    在编写一个socket程序时使用了如下代码 但在执行时却发生异常 意思就是说没办法分配获取到的IP地址,然后debug了一下获取到的IP地址以及主机查询到的IP地址果然存在冲突 这个时候出现了有意思的地方,为什么代码获取的仍然是我修改ip之前的IP地址4.222,而不是修改之后的

    2024年02月12日
    浏览(30)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包