Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

news/2024/12/14 13:27:45/文章来源:https://www.cnblogs.com/xixixing/p/18606604

错误报在了forward里的Conv2d处。原因是函数写在forward里可能默认cpu,如果写在init构造函数里,就不需要再指定cuda。

修改为箭头指示就不再报错了。

 

【参考】

Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same-CSDN博客

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/852493.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

jquery半透明拖拽窗口插件

这是一款jquery半透明拖拽窗口插件。该插件可以在页面生成可以拖拽、最大化、最小化的浮动窗口。在线演示 下载使用方法 在页面中引入style.css、jquery和jquery-translucent.js文件。<link rel="stylesheet" type="text/css" href="style.css&quo…

Marvelous Designer高版本更改界面字体大小

打开软件 打开 设置/用户自定义 - 用户自定义选择用户界面 - 显示 - 自动规模不勾选 - 分辨率选择大重启软件即可

golang:第三方库:用jordan-wright/email发送邮件

一,安装第三方库: $ go get -u github.com/jordan-wright/email go: downloading github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible go: added github.com/jordan-wright/email v4.0.1-0.20210109023952-943e75fe5223+incompatible 二,代…

为了改一行代码,我花了10多天时间,让性能提升了40多倍---Pascal架构GPU在vllm下的模型推理优化

ChatGPT生成的文章摘要 这篇博客记录了作者在家中使用Pascal显卡运行大型模型时遇到的挑战和解决方案。随着本地大型模型性能的提升,作者选择使用vllm库进行推理。然而,作者遇到了多个技术难题,需要自行编译vllm和PyTorch,以支持Pascal架构的显卡。编译过程中,作者深入研究…

vxe-table 实现复选框分页跨页勾选

vxe-table 实现复选框分页跨页勾选 官网:https://vxetable.cn 当使用数据分页与复选框多页勾选时,可以通过 checkbox-config.reserve 启用<template><div><vxe-button status="success" @click="getSelectEvent">获取已选</vxe-but…

键盘录入学生信息到集合

1.要求:思路:题目要求我们要录入三个学生对象,我们可以先完成一次学生录入,再使用三次循环即可完成三次学生录入 第一步,写好学生类 第二步,创建集合,用于储存学生对象第三步,使用键盘录入学生信息注意,因为有数字要录入,所以这里录入字符串要选择next() 第四步,创…

2024-2025-1 20241329 《计算机基础与程序设计》第十二周学习总结

作业信息 作业归属课程:2024-2025-1-计算机基础与程序设计 作业要求:2024-2025-1计算机基础与程序设计第十二周作业 作业目标:《C语言程序设计》第11章 作业正文:2024-2025-1 20241329 《计算机基础与程序设计》第十二周学习总结 教材学习内容总结 《C语言程序设计》第11章…

四六级规则及策略

四六级规则:1写作,2听力,3阅读和翻译满分710,加权后425分(也就是分数到百分之60)及格。加权,就是英语最后成绩不仅取决于卷面分还取决于在参考人群中的排名 策略: 先section C再翻译,最后section B。section A随便蒙。

阅记-横向优化-CA-《HPC-一文彻底搞懂并发编程与内存屏障》intel

目录Momory Ordering9.2.3.2 Neither Loads Nor Stores Are Reordered9.2.3.3 Stores Are Not Reordered With Earlier Loads9.2.3.4 Loads May Be Reordered with Earlier Stores (intel x64架构下唯一会有memory reorder的情况)内存屏障的实现 参考:HPC(高性能计算第一篇):一…