数据加载错误导致
可能像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()