pytorch dataloader 取batch_size时候出现bug的解决方式
1、
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 342 and 281 in dimension 3 at /pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1333
2、
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 1 in dimension 1
bug1 修改方法是将图片resize 成一个尺度：
解决1：
如果用transforms来resize ，关于尺寸的transform就是transforms.Resize(input_size, interpolation=3)了。
查看Resize的文档:
size (sequence or int) – Desired output size. If size is a sequence like (h, w), output size will be matched to this.
If size is an int, smaller edge of the image will be matched to this number. i.e, if height > width, then image will be rescaled to (size * height / width, size)
size这个参数既可以是一个数字，又可以是一个tuple，如果是数字，图片处理后的输出尺寸每次都会被重新计算为(input_size*height/width, size)，这样如果输入图片尺寸不一致，输出图片尺寸会不一致， 将导致bug1 问题
解决2：
也可以直接用opencv resize
以上这篇pytorch dataloader 取batch_size时候出现bug的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持小牛知识库。
-
本文向大家介绍写出几种IE6 bug的解决方法相关面试题，主要包含被问及写出几种IE6 bug的解决方法时的应答技巧和注意事项，需要的朋友参考一下 1）png24位的图片在iE6浏览器上出现背景，解决方案是做成PNG8.也可以引用一段脚本处理. 2）IE6双倍边距bug：在该元素中加入display:inline 或 display:block 3）像素问题 使用多个float和注释引起的 使用d
-
本文向大家介绍iOS 11 BUG的发现、定位和解决，包括了iOS 11 BUG的发现、定位和解决的使用技巧和注意事项，需要的朋友参考一下 前言 在iOS 11发布之后，出现了一系列适配相关的问题，UIScrollView在pagingEnabled=YES时滑动手势不灵敏，UITableView的滑动删除功能变动，UIImagePickerViewController的取消按钮点击区域变小等，本
-
本文向大家介绍Javascript 运动中Offset的bug解决方案，包括了Javascript 运动中Offset的bug解决方案的使用技巧和注意事项，需要的朋友参考一下 我们先来看看这个bug 是怎么产生的。 以下是用来测试的Javascript代码 ，目的是 让div慢慢变窄。 Javascript 代码 很简单，运行一下 没有任何问题，如愿的div在慢慢变小。 那这个offset 的bu
-
本文向大家介绍ztree获取选中节点时不能进入可视区域出现BUG如何解决，包括了ztree获取选中节点时不能进入可视区域出现BUG如何解决的使用技巧和注意事项，需要的朋友参考一下 zTree 是一个依靠 jQuery 实现的多功能 “树插件”。优异的性能、灵活的配置、多种功能的组合是 zTree 最大优点。 zTree 的特点编辑 ● zTree v3.0 将核心代码按照功能进行了分割，不需要的代
-
本文向大家介绍写出你遇到过IE6/7/8/9的BUG及解决方法相关面试题，主要包含被问及写出你遇到过IE6/7/8/9的BUG及解决方法时的应答技巧和注意事项，需要的朋友参考一下 把以前兼容IE6.7学习的东西搬出来了，还以为不见了。 兼容性问题 1.IE6margin双边距问题 2.IE67 li间隙问题 3.图片间隙问题——vertical-align：top 4.ie6下高度小于19px处理
-
本文向大家介绍解决Python出现_warn_unsafe_extraction问题的方法，包括了解决Python出现_warn_unsafe_extraction问题的方法的使用技巧和注意事项，需要的朋友参考一下 在Python项目中运行出现了“AttributeError: ResourceManager instance has no attribute ‘_warn_unsafe_extr