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

文三路张同学

于 2023-02-24 21:55:04 发布

阅读量86

点赞数

分类专栏：我的科研之路~ 其他文章标签： python 深度学习机器学习

本文链接：https://blog.csdn.net/qq_36160277/article/details/129209363

版权

我的科研之路~ 同时被 2 个专栏收录

46 篇文章 2 订阅

订阅专栏

其他

26 篇文章 0 订阅

订阅专栏

如题。
记录一下自己遇到的问题：
在这里插入图片描述

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