數(shù)據(jù)加載錯(cuò)誤導(dǎo)致
可能像https://blog.csdn.net/weixin_41278720/article/details/84586734一樣的問(wèn)題，但是這里沒(méi)問(wèn)題

我又查看了我的transform，由于沒(méi)有resize導(dǎo)致報(bào)錯(cuò)
train_transform = transforms.Compose([
#transforms.Resize((32, 32)),
transforms.RandomCrop(32, padding=4), # 將圖像轉(zhuǎn)化為32 * 32
transforms.RandomHorizontalFlip(), # 隨機(jī)水平翻轉(zhuǎn)
transforms.RandomRotation(degrees=15), # 隨機(jī)旋轉(zhuǎn)，角度15
transforms.ToTensor(), # 將numpy數(shù)據(jù)類型轉(zhuǎn)化為Tensor，torch可接受的類型
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) # 歸一化

總之就是數(shù)據(jù)加載的維度問(wèn)題

AttributeError: ‘StepLR’ object has no attribute 'get_last_lr’
查閱資料得到版本有點(diǎn)低，這是pytorch以后版本的attribute，如果用這個(gè)版本（1.2.0）換成
.get_lr()

總結(jié)

以上是生活随笔為你收集整理的pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。