模型训练测试技巧记录

我是一个粗心的人,有时候会犯一些低级的错误。低级错误常常使人懊恼,我想通过养成好的习惯可以避免,逐渐改善因为低级错误的时间浪费。

1、写一个模型训练的bat文件,到训练模型的时候直接稍作路径修改就可以直接用了。

例如要训练一个lenet模型,我有4块显卡,我想试试不同的训练策略,那我分别run4个bat就可以。这个bat应该是包含lmdb或者leveldb数据格式的生成,mean的生成,以及最后的train。

2、模型的log分析。

caffe自带的工具:caffe-master/tools/extra/parse_log.sh caffe-master/tools/extra/extract_seconds.py caffe-master/tools/extra/plot_training_log.py.example。

具体的使用方法看这篇博客:https://www.zhihu.com/question/49521165/answer/127675889

怎样在windows下输出训练caffemodel的log日志并画出accuracy和loss曲线?

3、deploy的prototxt对应的层名一定要对训练模型的一致。

不然加载caffemodel时就会出错或者是某个层被ignore,参数没有复制而是随机化了,这样测试结果就会有很大的错误。我犯了一个这样的错误。

先写这么多,我再遇到问题时慢慢补充。



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值