diff --git a/tools/experiment.py b/tools/experiment.py index dec87f42336..abae60a9f40 100644 --- a/tools/experiment.py +++ b/tools/experiment.py @@ -725,7 +725,7 @@ def _prepare_run_command(self, command: List[str]) -> bool: output_path = str(self._workspace / "outputs" / "latest_trained_model") self.set_arguments_to_cmd(command, "--output", output_path) elif cmd_entry == "optimize": - if previous_cmd == "export": + if previous_cmd == "export": # execute PTQ. If not, execute QAT file_path = self._find_model_path(previous_cmd) if file_path is None: return False