模型正常进行训练，训练好后想使用一个文本进行一下测试，发现居然在forward函数报错。
开始百思不得其解，训练都是正常的，按evaluate函数的逻辑写了测试样例，怎么会报错呢。
报错如下图：
最后经过仔细的检查，发现是forward()函数里squeeze()函数的问题，这个函数将某些满足只有一个元素的维度删除没了。因此解决办法是使用两个测试样本封装入一个batch中，问题解决了。
还是要更细心
| |
|
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
| -> 人工智能 -> pytorch Dimension out of range (expected to be in range of [-2 1] but got 2)问题记录 -> 正文阅读 |
|
|
[人工智能]pytorch Dimension out of range (expected to be in range of [-2 1] but got 2)问题记录 |
|
模型正常进行训练，训练好后想使用一个文本进行一下测试，发现居然在forward函数报错。 还是要更细心 |
|
|
|
|
| 上一篇文章 下一篇文章 查看所有文章 |
|
|
开发:
C++知识库
Java知识库
JavaScript
Python
PHP知识库
人工智能
区块链
大数据
移动开发
嵌入式
开发工具
数据结构与算法
开发测试
游戏开发
网络协议
系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程 数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁 |
| 360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年5日历 | -2024/5/21 17:35:55- |
|
| 网站联系: qq:121756557 email:121756557@qq.com IT数码 |