用训练好的模型生成数据，偶尔会报下边的错，位置不一样，跟您请教一下，怎么解决呢？网上查的方法都不行
kavinwow100 opened this issue · 0 comments
kavinwow100 commented
The size of tensor a (400) must match the size of tensor b (401) at non-singleton dimension 3
CUDA error: device-side assert triggered