pytorch数据预处理错误的解决


Posted in Python onFebruary 20, 2020

出错:

Traceback (most recent call last):
 File "train.py", line 305, in <module>
 train_model(model_conv, criterion, optimizer_conv, exp_lr_scheduler)
 File "train.py", line 145, in train_model
 for inputs, age_labels, gender_labels in dataloaders[phase]:
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 637, in __next__
 return self._process_next_batch(batch)
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 658, in _process_next_batch
 raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 138, in _worker_loop
 samples = collate_fn([dataset[i] for i in batch_indices])
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 232, in default_collate
 return [default_collate(samples) for samples in transposed]
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 232, in <listcomp>
 return [default_collate(samples) for samples in transposed]
 File "/home/home/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 209, in default_collate
 return torch.stack(batch, 0, out=out)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 224 and 228 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1307

这是因为输入的大小不匹配,跟数据集有关,也跟数据预处理中的函数相关:

transforms.Resize(input_size)

该函数是按比例缩放,可能是因为该数据集的分辨率不同,所以出来的结果不是(224,224)的,解决办法是改为使用:

transforms.Resize((input_size, input_size))

即可

以上这篇pytorch数据预处理错误的解决就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 判断一个进程是否存在
Apr 09 Python
Python下的Mysql模块MySQLdb安装详解
Apr 09 Python
python使用xlrd实现检索excel中某列含有指定字符串记录的方法
May 09 Python
Python 实现简单的电话本功能
Aug 09 Python
Android基于TCP和URL协议的网络编程示例【附demo源码下载】
Jan 23 Python
Python简单实现两个任意字符串乘积的方法示例
Apr 12 Python
详解Ubuntu16.04安装Python3.7及其pip3并切换为默认版本
Feb 25 Python
Python实现插入排序和选择排序的方法
May 12 Python
Python 旋转打印各种矩形的方法
Jul 09 Python
Pytorch使用MNIST数据集实现基础GAN和DCGAN详解
Jan 10 Python
一篇文章带你搞定Ubuntu中打开Pycharm总是卡顿崩溃
Nov 02 Python
教你漂亮打印Pandas DataFrames和Series
May 29 Python
Python异常继承关系和自定义异常实现代码实例
Feb 20 #Python
Python安装与卸载流程详细步骤(图解)
Feb 20 #Python
PyCharm 专业版安装图文教程
Feb 20 #Python
Python3.6安装卸载、执行命令、执行py文件的方法详解
Feb 20 #Python
Python 实现黑客帝国中的字符雨的示例代码
Feb 20 #Python
浅谈spring boot 集成 log4j 解决与logback冲突的问题
Feb 20 #Python
Python OpenCV读取显示视频的方法示例
Feb 20 #Python
You might like
用PHP函数解决SQL injection
2006/12/09 PHP
php simplexmlElement操作xml的命名空间实现代码
2011/01/04 PHP
php删除指定目录的方法
2015/04/03 PHP
浅谈PHP的exec()函数无返回值排查方法(必看)
2017/03/31 PHP
PHP类的自动加载与命名空间用法实例分析
2020/06/05 PHP
Javascript学习笔记6 prototype的提出
2010/01/11 Javascript
腾讯UED 漂亮的提示信息效果代码
2011/09/12 Javascript
JS防止用户多次提交的简单代码
2013/08/01 Javascript
Jquery同辈元素选中/未选中效果的实例代码
2013/08/01 Javascript
使用JS+plupload直接批量上传图片到又拍云
2014/12/01 Javascript
javascript中定义类的方法详解
2015/02/10 Javascript
个人总结的一些JavaScript技巧、实用函数、简洁方法、编程细节
2015/06/10 Javascript
基于javascript实现checkbox复选框实例代码
2016/01/28 Javascript
详解利用jsx写vue组件的方法示例
2017/07/17 Javascript
基于Vue框架vux组件库实现上拉刷新功能
2017/11/28 Javascript
利用React高阶组件实现一个面包屑导航的示例
2020/08/23 Javascript
EXTJS7实现点击拖拉选择文本
2020/12/17 Javascript
[05:29]2014DOTA2国际邀请赛 赛后专访:LGDNewbee顺利过关
2014/07/13 DOTA
Python Web框架Flask下网站开发入门实例
2015/02/08 Python
Python冒泡排序注意要点实例详解
2016/09/09 Python
python去掉 unicode 字符串前面的u方法
2018/10/21 Python
解决pytorch 交叉熵损失输出为负数的问题
2020/07/07 Python
torchxrayvision包安装过程(附pytorch1.6cpu版安装)
2020/08/26 Python
python+selenium爬取微博热搜存入Mysql的实现方法
2021/01/27 Python
Flask中jinja2的继承实现方法及实例
2021/03/03 Python
Nili Lotan官网:Nili Lotan同名品牌
2018/01/07 全球购物
海滩咖啡馆:Beach Cafe
2018/02/02 全球购物
美国Randolph太阳镜官网:美国制造的飞行员太阳镜和射击眼镜
2018/06/15 全球购物
欧姆龙医疗欧洲有限公司:Omron Healthcare Europe B.V
2020/06/13 全球购物
服装销售人员求职自我评价
2013/09/26 职场文书
缴纳养老保险的证明
2014/01/10 职场文书
材料工程专业毕业生求职信
2014/03/04 职场文书
大学理论知识学习自我鉴定
2014/04/28 职场文书
工商管理本科生求职信
2014/07/13 职场文书
Java生成读取条形码和二维码的简单示例
2021/07/09 Java/Android
Python超详细分步解析随机漫步
2022/03/17 Python