fix prof switch

7775abd7 · Hui Zhang · a40312b6 · 7775abd7
隐藏空白更改
内联并排

Showing with 1 addition and 3 deletion

deepspeech/training/trainer.py deepspeech/training/trainer.py +1 -3

未找到文件。
--- a/deepspeech/training/trainer.py
+++ b/deepspeech/training/trainer.py
@@ -200,10 +200,8 @@ class Trainer():
                batch_sampler.set_epoch(self.epoch)

    def after_train_batch(self):
-        if self.args.profiler_options:
-            profiler.add_profiler_step(self.args.profiler_options)
-
        if self.args.benchmark_max_step and self.iteration > self.args.benchmark_max_step:
+            profiler.add_profiler_step(self.args.profiler_options)
            logger.info(
                f"Reach benchmark-max-step: {self.args.benchmark_max_step}")
            sys.exit(