当前位置: 首页 > news >正文

网站页面风格分类邢台开发区网站

网站页面风格分类,邢台开发区网站,搜索引擎有哪些软件,装修报价器文章目录 微调总结 微调代码实现 微调 总结 微调通过使用在大数据上的恶道的预训练好的模型来初始化模型权重来完成提升精度。预训练模型质量很重要微调通常速度更快、精确度更高 微调代码实现 导入相关库 %matplotlib inline import os import torch import torchvision f… 文章目录 微调总结 微调代码实现 微调 总结 微调通过使用在大数据上的恶道的预训练好的模型来初始化模型权重来完成提升精度。预训练模型质量很重要微调通常速度更快、精确度更高 微调代码实现 导入相关库 %matplotlib inline import os import torch import torchvision from torch import nn from d2l import torch as d2l import matplotlib as plt获取数据集 d2l.DATA_HUB[hotdog] (d2l.DATA_URL hotdog.zip,fba480ffa8aa7e0febbb511d181409f899b9baa5)data_dir d2l.download_extract(hotdog)train_imgs torchvision.datasets.ImageFolder(os.path.join(data_dir,train)) test_imgs torchvision.datasets.ImageFolder(os.path.join(data_dir,test))print(train_imgs) print(train_imgs[0]) train_imgs[0][0]查看数据集中图像的形状 hotdogs [train_imgs[i][0] for i in range(8)] not_hotdogs [train_imgs[-i-1][0] for i in range(8)] d2l.show_images(hotdogs not_hotdogs, 2 ,8, scale1.4)数据增强 # 图像增广 normalize torchvision.transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224,0.225] ) train_augs torchvision.transforms.Compose( # 训练集数据增强[torchvision.transforms.RandomResizedCrop(224),torchvision.transforms.RandomHorizontalFlip(),torchvision.transforms.ToTensor(),normalize] ) test_augs torchvision.transforms.Compose( # 验证集不做数据增强[torchvision.transforms.Resize(256),torchvision.transforms.CenterCrop(224),torchvision.transforms.ToTensor(),normalize] )定义和初始化模型 # 下载resnet18 # 老pretrainTrue: 也下载预训练的模型参数 # 新weightstorchvision.models.ResNet18_Weights.IMAGENET1K_V1 pretrained_net torchvision.models.resnet18(weightstorchvision.models.ResNet18_Weights.IMAGENET1K_V1)print(pretrained_net.fc)微调模型 (1)直接修改网络层如最后全连接层512—1000改成512—2(2)在增加一层分类层如512—1000, 改成512—1000, 1000—2 本次选择1将resnet18最后全连接层的输出改成自己训练集的类别并初始化最后全连接层的权重参数 finetune_net pretrained_net finetune_net.fc nn.Linear(finetune_net.fc.in_features, 2) nn.init.xavier_uniform_(finetune_net.fc.weight)print(finetune_net)训练模型 特征提取层预训练层使用较小的学习率输出全连接层微调层使用较大的学习率 def train_fine_tuning(net, learning_rate, batch_size128, num_epochs10, param_groupTrue):train_iter torch.utils.data.DataLoader(torchvision.datasets.ImageFolder(os.path.join(data_dir,train), transformtrain_augs),batch_sizebatch_size,shuffleTrue)test_iter torch.utils.data.DataLoader(torchvision.datasets.ImageFolder(os.path.join(data_dir, test), transformtest_augs),batch_sizebatch_size)device d2l.try_all_gpus()loss nn.CrossEntropyLoss(reductionnone)if param_group:params_1x [param for name, param in net.named_parameters()if name not in [fc.weight, fc.bias]]trainer torch.optim.SGD([{params: params_1x}, {params: net.fc.parameters(), lr: learning_rate * 10}],lrlearning_rate, weight_decay0.001)else:trainer torch.optim.SGD(net.parameters(),lrlearning_rate,weight_decay0.001)d2l.train_ch13(net, train_iter, test_iter, loss,trainer, num_epochs, device)训练模型 import time# 在开头设置开始时间 start time.perf_counter() # start time.clock() python3.8之前可以train_fine_tuning(finetune_net, 5e-5, 128, 10)# 在程序运行结束的位置添加结束时间 end time.perf_counter() # end time.clock() python3.8之前可以# 再将其进行打印即可显示出程序完成的运行耗时 print(f运行耗时{(end-start):.4f} s)直接训练整个模型都使用相同的学习率重新训练 scracth_net torchvision.models.resnet18() scracth_net.fc nn.Linear(scracth_net.fc.in_features, 2)import time# 在开头设置开始时间 start time.perf_counter() # start time.clock() python3.8之前可以train_fine_tuning(scracth_net, 5e-4, param_groupFalse)# 在程序运行结束的位置添加结束时间 end time.perf_counter() # end time.clock() python3.8之前可以# 再将其进行打印即可显示出程序完成的运行耗时 print(f运行耗时{(end-start):.4f} s)
http://www.sadfv.cn/news/407545/

相关文章:

  • wordpress网站go.php跳转筑建网
  • 一个网站 两个数据库西安市今天发生的重大新闻
  • 网站建设销售好做吗辽宁建设工程信息网新域名
  • 网站做关键词首页火车头 wordpress 发布
  • 搭建网站挣钱哪些网站是单页面
  • 自己做的网站能放到阿里云上做地产的设计网站
  • 网站建设合同英文版自己免费制作app平台
  • 可植入代码网站开发域名有哪些
  • 做网站的人还能做什么网站视频管理系统
  • 网站建设实训心得与建议口碑营销成功案例有哪些
  • 网站设计配色怎么做塑料公司网站建设方案
  • 设计网站流程磁力猫引擎入口
  • 中国3.15诚信建设联盟网站为什么网站建设图片显示不出来
  • 怎么设计网站页面海北公司网站建设价格低
  • 网站建设团队成员wordpress点餐
  • 网站开发源代码修改建筑公司分几级资质
  • 华润置地建设事业部官方网站网站开发的需求文档模板
  • 商城网站开发需求分析wordpress 变更中文
  • 简单的asp网站建设人行官方网站
  • 邯郸手机网站建设费用网站定制开发优点
  • 做特卖的网站上品折扣做网站的图片Pc端和手机端的区别
  • 奇迹网站建设多少钱贵阳网站定制
  • 公司长沙建站网站建设的流程电子商务
  • 自己网站建设的流程是什么wordpress 主题2周前
  • 西安网站建设公司咪豆中国电信网上营业厅
  • 怎么做点击图片进网站7k7k网页游戏入口
  • 南京华夏天成建设有限公司网站怎么可以创建网站
  • 查企业营业执照的网站自己做网站需要买哪些
  • 京东网站建设的基本情况佛山新网站建设策划
  • 龙华网站建设专业定制企业重庆网站制作公司