parameters: (self, sess, network, imdb, roidb, output_dir, pretrained_model=None, pretrained_ckpt=None)
docstring: Initialize the SolverWrapper.
docstring_summary: Initialize the SolverWrapper.
docstring_tokens: Initialize the SolverWrapper .
function: def __init__(self, sess, network, imdb, roidb, output_dir, pretrained_model=None, pretrained_ckpt=None):
          """Initialize the SolverWrapper."""
          self.net = network
          self.imdb = imdb
          self.roidb = roidb
          self.output_dir = output_dir
          self.pretrained_model = pretrained_model
          self.pretrained_ckpt = pretrained_ckpt
  
          # For checkpoint
          self.saver = tf.train.Saver(tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES), max_to_keep=12)
function_tokens: def __init__ ( self , sess , network , imdb , roidb , output_dir , pretrained_model = None , pretrained_ckpt = None ) : self . net = network self . imdb = imdb self . roidb = roidb self . output_dir = output_dir self . pretrained_model = pretrained_model self . pretrained_ckpt = pretrained_ckpt # For checkpoint self . saver = tf . train . Saver ( tf . get_collection ( tf . GraphKeys . GLOBAL_VARIABLES ) , max_to_keep = 12 )