Searched refs:include_top (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/examples/export_models/models/ |
| D | mini_alexnet.py | 34 def __init__(self, num_classes=10, channel=1, phase='train', include_top=True): argument
40 self.include_top = include_top
41 if self.include_top:
59 if not self.include_top:
|
| /third_party/mindspore/tests/st/summary/ |
| D | test_summary_ops.py | 37 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
43 self.include_top = include_top
44 if self.include_top:
65 if not self.include_top:
|
| D | test_summary_collector.py | 51 def __init__(self, num_class=10, num_channel=1, include_top=True): argument
57 self.include_top = include_top
58 if self.include_top:
83 if not self.include_top:
|