(self, phase_name, step)
| 120 | _step_result = False |
| 121 | |
| 122 | def step_complete(self, phase_name, step): |
| 123 | try: |
| 124 | if phase_name in self._active_phases.keys(): |
| 125 | self._active_phases[phase_name]["tq"].update( |
| 126 | step - self._active_phases[phase_name]["tq"].n |
| 127 | ) |
| 128 | return self._step_result |
| 129 | except KeyboardInterrupt: |
| 130 | # There is no need to propagate this exception to TensorRT. We can simply cancel the build. |
| 131 | return False |
| 132 | |
| 133 | |
| 134 | class Engine: |