解决报错:invalid argument 0: Sizes of tensors must match except in dimension 0.
发布日期:2021-07-01 04:36:54
浏览次数:3
分类:技术文章
本文共 1553 字,大约阅读时间需要 5 分钟。
报错如下:
Traceback (most recent call last): File “6_database_deal_.py”, line 73, in for i, data in enumerate(test_loader): File “/home/muli/anaconda3/lib/python3.5/site-packages/torch/utils/data/dataloader.py”, line 560, in next batch = self.collate_fn([self.dataset[i] for i in indices]) File “/home/muli/anaconda3/lib/python3.5/site-packages/torch/utils/data/_utils/collate.py”, line 68, in default_collate return [default_collate(samples) for samples in transposed] File “/home/muli/anaconda3/lib/python3.5/site-packages/torch/utils/data/_utils/collate.py”, line 68, in return [default_collate(samples) for samples in transposed] File “/home/muli/anaconda3/lib/python3.5/site-packages/torch/utils/data/_utils/collate.py”, line 43, in default_collate return torch.stack(batch, 0, out=out) RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 309 and 580 in dimension 2 at /pytorch/aten/src/TH/generic/THTensor.cpp:711数据集图像大小不一,加载训练集时进行了RandomResizedCrop , 但是在测试时忘记了,因此出现了以下报错信息:
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 309 and 580 in dimension 2 at /pytorch/aten/src/TH/generic/THTensor.cpp:711解决办法:
testTransform部分加入 transforms.Resize((224, 224))# 训练trainTransform = transforms.Compose([ transforms.RandomResizedCrop(224), # 随机裁剪, transforms.RandomHorizontalFlip(), # 随机水平翻转 transforms.ToTensor(), normTransform # 正则化])# 测试testTransform = transforms.Compose([ transforms.Resize((224, 224)), # 调整图像大小 transforms.ToTensor(), normTransform # 正则化])
转载地址:https://mymuli.blog.csdn.net/article/details/100566783 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
哈哈,博客排版真的漂亮呢~
[***.90.31.176]2024年05月06日 18时44分18秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
(12) Hadoop 升级技能
2019-05-06
(总结)Nginx配置文件nginx.conf中文详解
2019-05-06
wget 下载
2019-05-06
HTTPS的七个误解
2019-05-06
ssl证书配置
2019-05-06
为什么使用<!DOCTYPE HTML>
2019-05-06
leetcode 477 Total Hamming Distance C++
2019-05-06
笑话收集
2019-05-06
超级程序员神话
2019-05-06
advanced installer制作windows软件安装界面
2019-05-06
操作系统读书笔记
2019-05-06
2017年计划
2019-05-06
【转】电容滤波的两个要点_Detective_ALong_新浪博客
2019-05-06
html5星星评分组件,sass双循环写法
2019-05-06
h5锁屏提醒-锁横屏和锁竖屏
2019-05-06
LeetCode 方法详解
2019-05-07
View事件分发情况总结
2019-05-07
运放(二):运放带宽
2019-05-07
activiti入门列子一个简单的activiti请假流程
2019-05-07