(self, name, nets=None, num_iter=None)
| 2687 | return name |
| 2688 | |
| 2689 | def __init__(self, name, nets=None, num_iter=None): |
| 2690 | self._step = caffe2_pb2.ExecutionStep() |
| 2691 | self._step.name = name or ExecutionStep._get_next_step_name('step') |
| 2692 | self._net_dict = OrderedDict() |
| 2693 | self._is_used = False |
| 2694 | self._substeps = [] |
| 2695 | if nets is not None: |
| 2696 | if type(nets) is Net: |
| 2697 | nets = [nets] |
| 2698 | for net in nets: |
| 2699 | if _add_net_to_dict(self._net_dict, net): |
| 2700 | self._step.network.extend([get_net_name(net)]) |
| 2701 | if num_iter is not None: |
| 2702 | self._step.num_iter = num_iter |
| 2703 | |
| 2704 | def get_net(self, name): |
| 2705 | return self._net_dict[name] |
nothing calls this directly
no test coverage detected