pytorch dataloader取batch_size时候出现错误的解决方式

  

1,   

  

RuntimeError:无效的论点0:张量的大小必须匹配除了维0。了342年和281年在维3/pytorch/阿托恩/src/TH/一般/THTensorMoreMath.cpp: 1333

     

2,   

  

RuntimeError:无效的论点0:张量的大小必须匹配除了维0。3和1维1

     

  

<>强解决1:

  

如果用变换来调整,关于尺寸的变换就是。调整(input_size插值=3)了。

  

查看大小的文档:

  
  

大小(序列或int)——期望输出值大小。如果大小是一个序列(h, w),输出尺寸将匹配。

  

如果int大小,小的边缘图像匹配这个数字。我。e,如果高度比;宽度,然后将新图像(*高度/宽度大小,大小)

     

大小这个参数既可以是一个数字,又可以是一个元组,如果是数字,图片处理后的输出尺寸每次都会被重新计算为(input_size *高度/宽度、大小),这样如果输入图片尺寸不一致,输出图片尺寸会不一致,将导致bug1问题

  

<>强解决2:

  

  

以上这篇pytorch dataloader取batch_size时候出现错误的解决方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

pytorch dataloader取batch_size时候出现错误的解决方式