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