必备的文件

  • solver.prototxt 训练求解的配置文件

    > 此文件用于训练神经网络时配置网络该如何学习。
    
  • train-test.prototxt 网络定义文件,也可以分开定义

    > 此文件用于定义训练或者测试使用的神经网络
    
  • xxx.caffemodel 训练完成之后得到的神经网络权重文件

    > 用这个文件可以重载训练之后得到的网络权重,同 deploy 网络一起用于应用
    
  • deploy.prototxt 应用时实用的网络

solver.prototxt 文件格式

  • iter (iteration) 对应一次权限更新。也就是说一次前向和一次后向,然后更新权重。
  • max_iter 本次训练的目标,权重更新多少次。max_iter * batch_size / data.length = epoch。 一般来说 epoch 等于 10 差不多合适。

train/test/deploy.prototxt 文件格式


Airy
2.4k 声望69 粉丝

github.com/airylinus


引用和评论

0 条评论