zoukankan      html  css  js  c++  java
  • caffe solver.prototxt 生成

    from caffe.proto import caffe_pb2
    
    s = caffe_pb2.SolverParameter()
    
    path='/home/xxx/data/'
    solver_file=path+'solver.prototxt'     #solver文件保存位置
    
    s.train_net = path+'train.prototxt'     # 训练配置文件
    s.test_net.append(path+'val.prototxt')  # 测试配置文件
    s.test_interval = 782                   # 测试间隔
    s.test_iter.append(313)                 # 测试迭代次数
    s.max_iter = 78200                      # 最大迭代次数
    
    s.base_lr = 0.001                       # 基础学习率
    s.momentum = 0.9                        # momentum系数
    s.weight_decay = 5e-4                   # 权值衰减系数
    s.lr_policy = 'step'                    # 学习率衰减方法
    s.stepsize=26067                        # 此值仅对step方法有效
    s.gamma = 0.1                           # 学习率衰减指数
    s.display = 782                         # 屏幕日志显示间隔
    s.snapshot = 7820
    s.snapshot_prefix = 'shapshot'
    s.type = “SGD”                          # 优化算法
    s.solver_mode = caffe_pb2.SolverParameter.GPU
    
    with open(solver_file, 'w') as f:
        f.write(str(s))
  • 相关阅读:
    Python格式化字符 %s %d %f
    FTP学习笔记
    万维网
    TCP笔记
    TCP流量控制
    笔记传输层
    传输层协议
    网络层
    以太网笔记
    计算机网络物理层
  • 原文地址:https://www.cnblogs.com/houjun/p/9819989.html
Copyright © 2011-2022 走看看