pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0
數據加載錯誤導致
可能像https://blog.csdn.net/weixin_41278720/article/details/84586734一樣的問題,但是這里沒問題
我又查看了我的transform,由于沒有resize導致報錯
train_transform = transforms.Compose([
#transforms.Resize((32, 32)),
transforms.RandomCrop(32, padding=4), # 將圖像轉化為32 * 32
transforms.RandomHorizontalFlip(), # 隨機水平翻轉
transforms.RandomRotation(degrees=15), # 隨機旋轉,角度15
transforms.ToTensor(), # 將numpy數據類型轉化為Tensor,torch可接受的類型
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) # 歸一化
總之就是數據加載的維度問題
AttributeError: ‘StepLR’ object has no attribute 'get_last_lr’
查閱資料得到版本有點低,這是pytorch以后版本的attribute,如果用這個版本(1.2.0)換成
.get_lr()
總結
以上是生活随笔為你收集整理的pytorch:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 世界级的客户服务:当今公司的正确行事
- 下一篇: 【区块链】复习九之以太坊数据结构