pytorch数据预处理错误的解决
2020-03-11 17:36
387 查看
出错:
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数据预处理错误的解决就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- PLSQL数据导入导出问题解决(空表、大字段表、表空间错误等)
- 【C#】解决进行反序列化时出错:。InnerException 消息是“反序列化对象 属于类型 System.String 时出现错误。读取 XML 数据时,超出最大字符串内容长度配额 (8192)。
- SSM框架返回JSON数据,报405、406错误的解决方法
- MySQL 数据批量恢复时 [Err] 2006 - MySQL server has gone away 错误的解决
- 解决SQL将varchar值转换为数据类型为int的列时发生语法错误
- SQL Server导入导出数据时最常见的一个错误解决方法
- oracle中使用impdp数据泵导入数据提示“ORA-31684:对象类型已经存在”错误的解决
- Restful框架中导入数据库表,选择Application/Json选项显示数据时出现500错误的解决办法
- 解决SPS平台列表视图切换为数据表模式错误解决方法
- mysql导入数据时提示 USING BTREE 错误解决办法
- mysql导入数据时 USING BTREE 错误解决办法
- mvc数据表机构修改引起运行错误的解决方法
- 删除文件出现 数据错误 循环冗余的解决办法
- sql2005 数据复制《发布 '%1' 的并发快照不可用》错误的解决办法
- Java 多线程 死锁 隐性死锁 数据竞争 恶性数据竞争 错误解决深入分析 全方向举例
- SpringMVC + Spring + MyBatis 学习笔记:提交数据遭遇基础类型和日期类型报400错误解决方法
- Windows2000 在保存IP安全数据时出现拒绝访问错误80070005的解决方法
- SpringMVC中出现" 400 Bad Request "错误(用@ResponseBody处理ajax传过来的json数据转成bean)的解决方法
- 用sql2005导出数据到access或excel时,出现" 找不到sourceconnectionoledb"的错误 (已解决)
- 解决SSD-Tensorflow训练自己数据的InvalidArgumentError 错误