Update model.py

10 months ago · 1e5703fc6e
parent 0fe081aff2
commit 1e5703fc6e
1 changed files with 2 additions and 0 deletions
--- a/paddlespeech/s2t/exps/deepspeech2/model.py
+++ b/paddlespeech/s2t/exps/deepspeech2/model.py
@ -641,6 +641,8 @@ class DeepSpeech2ExportTester(DeepSpeech2Tester):
                self.args.export_path + ".pdmodel",
                self.args.export_path + ".pdiparams")
        deepspeech_config.disable_mkldnn()
        if (os.environ['CUDA_VISIBLE_DEVICES'].strip() != ''):
            deepspeech_config.enable_use_gpu(100, 0)
            deepspeech_config.enable_memory_optim()