Sets swish function as memory efficient (for training) or standard (for export)
(self, memory_efficient=True)
| 162 | self._swish = MemoryEfficientSwish() |
| 163 | |
| 164 | def set_swish(self, memory_efficient=True): |
| 165 | """Sets swish function as memory efficient (for training) or standard (for export)""" |
| 166 | self._swish = MemoryEfficientSwish() if memory_efficient else Swish() |
| 167 | for block in self._blocks: |
| 168 | block.set_swish(memory_efficient) |
| 169 | |
| 170 | |
| 171 | def extract_features(self, inputs): |
nothing calls this directly
no test coverage detected