pytorch 下载数据集

def get_data_set(args, train_flag=True):
    if train_flag:
        data_set = torchvision.datasets.__dict__[args.data_set]        
                     (root=args.data_path, train=True, 
                     transform=get_transformer(args.data_set, args.imsize,
                     args.cropsize, args.crop_padding, args.hflip), download=True)
    else:
        data_set = torchvision.datasets.__dict__[args.data_set] 
                    (root=args.data_path, train=False, 
                    transform=get_transformer(args.data_set), download=True)    
    return data_set

data_set = get_data_set(args, train_flag=True)