def __init__(self, prototxt):
"""Construct a Solver.
Parameters
----------
prototxt : str
The path of ``.prototxt`` file.
Returns
-------
Solver
The solver.
Examples
--------
>>> solver = Solver('solver.prototxt')
"""
self._param = pb.SolverParameter()
Parse(open(prototxt, 'r').read(), self._param)
self.ParseUpdateParam()
self._net = None
self._test_nets = []
self._layer_blobs = []
self._iter = self._current_step = 0
self._optimizer = None
self.scalar_writer = sw.ScalarSummary() if root_solver() else None
self.InitTrainNet()
self.InitTestNets()
self.BuildNets()
def __init__(self, prototxt):
self._param = pb.SolverParameter()
Parse(open(prototxt, 'r').read(), self._param)
self._net = None; self._test_nets = []
self._iter = self._current_step = 0
self.train = self.tests = self.update = self._updater = None
self.scalar_writer = sw.ScalarSummary() if root_solver() else None
self._lr_blobs = []
self.InitTrainNet()
self.InitTestNets()
self.CheckUpdateParam()