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记录详细调用堆栈日志的方法
May 05 Python
Python2.x版本中基本的中文编码问题解决
Oct 12 Python
详解Python 协程的详细用法使用和例子
Jun 15 Python
python2与python3中关于对NaN类型数据的判断和转换方法
Oct 30 Python
如何使用Flask-Migrate拓展数据库表结构
Jul 24 Python
django框架面向对象ORM模型继承用法实例分析
Jul 29 Python
python3 实现的对象与json相互转换操作示例
Aug 17 Python
基于python调用psutil模块过程解析
Dec 20 Python
Python调用Windows命令打印文件
Feb 07 Python
python:删除离群值操作(每一行为一类数据)
Jun 08 Python
如何设置PyCharm中的Python代码模版(推荐)
Nov 20 Python
Python实现石头剪刀布游戏
Jan 20 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 has encountered an Access Violation 错误的解决方法
2010/01/17 PHP
PHP文章采集URL补全函数(FormatUrl)
2012/08/02 PHP
PHP UTF8中文字符截断函数代码
2012/09/11 PHP
将博客园(cnblogs.com)数据导入到wordpress的代码
2013/01/06 PHP
apache中为php 设置虚拟目录
2014/12/17 PHP
PHP创建word文档的方法(平台无关)
2016/03/29 PHP
php 猴子摘桃的算法
2017/06/20 PHP
PHP7.0连接DB操作实例分析【基于mysqli】
2019/09/26 PHP
js CSS操作方法集合
2008/10/31 Javascript
js 图片等比例缩放代码
2010/05/13 Javascript
JQUBAR1.1 jQuery 柱状图插件发布
2010/11/28 Javascript
JQuery实现展开关闭层的方法
2015/02/17 Javascript
JS原型与原型链的深入理解
2017/02/15 Javascript
微信小程序 参数传递实例代码
2017/03/20 Javascript
javascript浏览器用户代理检测脚本实现方法
2017/10/27 Javascript
three.js中文文档学习之如何本地运行详解
2017/11/20 Javascript
Javascript的this详解
2019/03/23 Javascript
vue-cli中使用高德地图的方法示例
2019/03/28 Javascript
vue实现将一个数组内的相同数据进行合并
2019/11/07 Javascript
vue prop属性传值与传引用示例
2019/11/13 Javascript
微信小程序图片自适应实现解析
2020/01/21 Javascript
javascript贪吃蛇游戏设计与实现
2020/09/17 Javascript
tornado框架blog模块分析与使用
2013/11/21 Python
Python实现的几个常用排序算法实例
2014/06/16 Python
python实现批量获取指定文件夹下的所有文件的厂商信息
2014/09/28 Python
Python itertools模块详解
2015/05/09 Python
一个基于flask的web应用诞生 记录用户账户登录状态(6)
2017/04/11 Python
TensorFlow实现创建分类器
2018/02/06 Python
python3.x上post发送json数据
2018/03/04 Python
安德玛加拿大官网:Under Armour加拿大
2019/10/02 全球购物
教师自我鉴定范文
2014/03/20 职场文书
《彭德怀和他的大黑骡子》教学反思
2014/04/12 职场文书
村安全生产责任书
2014/08/25 职场文书
教师党员个人整改措施材料
2014/09/16 职场文书
千万级用户系统SQL调优实战分享
2022/03/03 MySQL
springboot用户数据修改的详细实现
2022/04/06 Java/Android