caffe_solver.py 文件源码

python
阅读 21 收藏 0 点赞 0 评论 0

项目:Triplet_Loss_SBIR 作者: TuBui 项目源码 文件源码
def add_params(self,params):
    """
    Set or update solver parameters
    """
    paramstr = ''
    for key, val in params.items():
      self.sp.ClearField(key) #reset field
      if isinstance(val,str):     #if val is a string
        paramstr += (key + ': ' + '"' + val + '"' + '\n')
      elif type(val) is list:     #repeatable field
        for it in val:
          paramstr += (key + ': ' + str(it) + '\n')
      elif type(val) == type(True): #boolean type
        if val:
          paramstr += (key + ': true\n')
        else:
          paramstr += (key + ': false\n')
      else:                       #numerical value
        paramstr += (key + ': ' + str(val) + '\n')
    #apply change
    text_format.Merge(paramstr, self.sp)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号