原文:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1 #805

具体错误日志如下: 解决办法: 你输入的图像数据的维度不完全是一样的,比如是训练的数据有 组,其中 组是 ,但有一组是 ,这样会导致Pytorch的检查程序报错 另外一个则是比较隐晦的batchsize的问题,Pytorch中检查你训练维度正确是按照每个batchsize的维度来检查的,比如你有 组数据 假设每组数据为三通道 px px的图像 ,batchsize为 ,那么每次训练则提取 , , ...

2020-01-25 01:38 0 1553 推荐指数:

查看详情

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM