2019年9月17日

解决:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 544 and 1935 in dimension 2 at ../aten/src/TH/generic/THTensor.cpp:711

摘要: 这种错误有两种可能: 1.你输入的图像数据的维度不完全是一样的,比如是训练的数据有100组,其中99组是256*256,但有一组是384*384,这样会导致Pytorch的检查程序报错 2.比较隐晦的batchsize的问题,Pytorch中检查你训练维度正确是按照每个batchsize的维度来检查 阅读全文

posted @ 2019-09-17 10:04 天道酬勤、 阅读(20464) 评论(0) 推荐(2) 编辑

导航