def __init__(self, name, nets, iterations=None):
self._step = caffe2_pb2.ExecutionStep()
self._step.name = name
if type(nets) is Net:
nets = [nets]
self._step.networks.extend([str(n) for n in nets])
if iterations:
self._step.iterations = iterations
def __init__(self, name, nets, num_iter=None):
self._step = caffe2_pb2.ExecutionStep()
self._step.name = name
if type(nets) is Net:
nets = [nets]
self._step.network.extend([str(n) for n in nets])
if num_iter is not None:
self._step.num_iter = num_iter
def __init__(self, name, nets=None, num_iter=None):
self._step = caffe2_pb2.ExecutionStep()
self._step.name = name
self._net_dict = OrderedDict()
self._is_used = False
self._substeps = []
if nets is not None:
if type(nets) is Net:
nets = [nets]
for net in nets:
if _add_net_to_dict(self._net_dict, net):
self._step.network.extend([get_net_name(net)])
if num_iter is not None:
self._step.num_iter = num_iter
def __init__(self, name): self._step = caffe2_pb2.ExecutionStep() self._step.name = name
