RuntimeError: The size of tensor a (30) must match the size of tensor b (54) at non-singleton dimension 0

问题: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数扩大为设定的数量

posted @ 2022-01-25 11:16  IamIron-Man  阅读(3766)  评论(0)    收藏  举报