首页 > 技术文章 > RuntimeError: The size of tensor a (30) must match the size of tensor b (54) at non-singleton dimension 0

IamIron-Man 2022-01-25 11:16 原文

问题:RT,下一组数据集的又变成RuntimeError: The size of tensor a (30) must match the size of tensor b (36) at non-singleton dimension 0,后面tensor b的值在不断变化

解决:全景分割里面初始设定的mask的数量是30个,数据集里的mask多于了30个,这时需要把初始mask数量调大

后续问题:初始设定的mask过大,会导致显存占用过大,因为它会让所有数据集的mask数扩大为设定的数量

推荐阅读