解决报错:invalid argument 0: Sizes of tensors must match except in dimension 0.
发布日期:2021-07-01 04:36:54
浏览次数:2
分类:技术文章
本文共 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 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
感谢大佬
[***.8.128.20]2024年04月26日 14时51分44秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
R语言实战
2019-05-07
MongoDB(一)
2019-05-07
MONGODB(二)——索引操作
2019-05-07
MONGODB(三)——Java操作Mongo
2019-05-07
数据挖掘(三)——决策树
2019-05-07
Vim——回顾整理
2019-05-07
R语言实战(三)——模拟随机游走数据
2019-05-07
贪心算法
2019-05-07
R语言实战(四)—— 基本数据管理
2019-05-07
R语言实战 —— 常见问题解决方法
2019-05-07
MyBatis——MyEclipse中使用mybatis-generator
2019-05-07
Redis学习(三) —— 持久化
2019-05-07
Java实现通过IP获取域名,通过域名获取IP
2019-05-07
Java位操作
2019-05-07
Redis学习(四) —— 管道 与 位图
2019-05-07
LeetCode 第 151 场周赛
2019-05-07
LeetCode 第 153 场周赛
2019-05-07
LeetCode设计实现题(一)
2019-05-07
LeetCode 第 154 场周赛
2019-05-07
图拉丁装机
2019-05-07