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()

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值