remve useless

pull/538/head
Hui Zhang 5 years ago
parent 4b6d4a1c06
commit 9a86f0672b

2
.gitignore vendored

@ -3,3 +3,5 @@
tools/venv
.vscode
*.log
*.pdmodel
*.pdiparams*

@ -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,

@ -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)

Loading…
Cancel
Save