參考鏈接:??????解決Pytorch dataloader時(shí)報(bào)錯(cuò)每個(gè)tensor維度不一樣的問(wèn)題_python_腳本之家
記錄一下自己遇到的bug:
問(wèn)題描述:?
問(wèn)題分析:
torch.stack(batch, 0, out=out)出錯(cuò),原因可能是:同一個(gè)batch的數(shù)據(jù)圖片的維度(H, W, C)要相同(可以見(jiàn)官方文檔:其shape必須一致)
問(wèn)題解決:文章來(lái)源:http://www.zghlxwxcb.cn/news/detail-610603.html
如果圖像尺寸不固定一致的話,那batch_size設(shè)置為1,問(wèn)題即可解決文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-610603.html
到了這里,關(guān)于RuntimeError: stack expects each tensor to be equal size, but got at entry的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!