首页 > 其他分享 >【debug】没有设置可训练参数时,训练集的准确率还在发生变化,可能是对训练集数据数据增强的原因。

【debug】没有设置可训练参数时,训练集的准确率还在发生变化,可能是对训练集数据数据增强的原因。

时间:2023-02-25 12:33:52浏览次数:41  
标签:Normalize ToTensor Compose 0.485 训练 准确率 transforms debug 224


如题。

记录一下自己遇到的问题:

【debug】没有设置可训练参数时,训练集的准确率还在发生变化,可能是对训练集数据数据增强的原因。_python

def build_transform(args):
if args.dataset == "flowers":
data_transforms = {
'train': transforms.Compose([
transforms.RandomRotation(30), # 数据增强
transforms.RandomResizedCrop(224), # 数据增强
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406],
[0.229, 0.224, 0.225])
]),
'valid': transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406],
[0.229, 0.224, 0.225])
]),
'test': transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize([0.485, 0.456, 0.406],
[0.229, 0.224, 0.225])
])
}

return data_transforms


标签:Normalize,ToTensor,Compose,0.485,训练,准确率,transforms,debug,224
From: https://blog.51cto.com/u_15346485/6085281

相关文章