训练方式调参技巧

一、模型选择

回归任务:人脸关键点检测,又要速度,又要精度

场景任务:

  1. 昆虫检测,不会瞬时移动,但是数量要清楚,所以应该选择精度比较高的模型。

paddlex调用模型的api:paddlex.det.模型名称,具体可以看api文档

  1. 人像分割,比如会议室换背景,需要速度块,但是精度可以不用太高PaddleX

  2. 文字识别,PaddleHub,能用就直接用,PaddleX用于训练的多

二、模型训练

常用优化器adam,GSD,也有超参数可以进行优化,比如学习率

使用paddlex进行训练的方式

  1. pdx.datasets.VOCDetection使用这个api进行数据处理,需要传数据集位置,训练列表和验证列表

  2. pdx.det.使用这个API进行模型加载,模型的输出类别数一定要改其他可以用默认的

  3. 使用刚刚的api创建出来的model使用train方法即可开始训练了,可以设置的参数有epoch和学习率等,具体可以查看对应的api文档。

三、超参优化

一般来说,超参数有:学习率,迭代次数,网络的层数,每层神经元的个数等等。

可以分为三类:网络结构(神经元数量),优化参数(优化器,学习率,衰减策略),正则化系数

学习率的调整太大,可能会不收敛,太小可能会训练得很慢,一般是使用warm_up的学习率衰减策略(刚开始设计的时候需要较大的学习率,随着学习的深入就逐渐下降学习率,但是不能小于0)

四、可视化结果

可以直接使用matplotelib查看预测结果,e还可以使用VisualDL对模型的训练loss进行可视化,还有不同学习率的影响对比图,以及模型的网络结构图。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值