diff --git a/.gitignore b/.gitignore index 24b4c4de6..93b7544a4 100644 --- a/.gitignore +++ b/.gitignore @@ -3,3 +3,5 @@ tools/venv .vscode *.log +*.pdmodel +*.pdiparams* diff --git a/deepspeech/exps/deepspeech2/model.py b/deepspeech/exps/deepspeech2/model.py index 020ed1b58..09280cf9f 100644 --- a/deepspeech/exps/deepspeech2/model.py +++ b/deepspeech/exps/deepspeech2/model.py @@ -305,15 +305,8 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): exit(-1) def export(self): - self.infer_model.eval() feat_dim = self.test_loader.dataset.feature_size - # static_model = paddle.jit.to_static( - # self.infer_model, - # input_spec=[ - # paddle.static.InputSpec(shape=[None, feat_dim, None], dtype='float32'), # audio, [B,D,T] - # paddle.static.InputSpec(shape=[None], dtype='int64'), # audio_length, [B] - # ]) paddle.jit.save( self.infer_model, self.args.export_path, diff --git a/deepspeech/training/trainer.py b/deepspeech/training/trainer.py index f93bc5754..a775aa6b5 100644 --- a/deepspeech/training/trainer.py +++ b/deepspeech/training/trainer.py @@ -267,10 +267,6 @@ class Trainer(): stream_handler.setFormatter(formatter) logger.addHandler(stream_handler) - # if not hasattr(self, 'output_dir'): - # self.logger = logger - # return - log_file = self.output_dir / 'worker_{}.log'.format(dist.get_rank()) # file_handler = logging.FileHandler(str(log_file)) # file_handler.setFormatter(formatter)