From 8d2f17672abf182ab332924e990eb648e03d7ea0 Mon Sep 17 00:00:00 2001 From: megemini Date: Tue, 3 Dec 2024 21:40:42 +0800 Subject: [PATCH] [Update] config disable_mkldnn --- paddlespeech/cls/exps/panns/deploy/predict.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/paddlespeech/cls/exps/panns/deploy/predict.py b/paddlespeech/cls/exps/panns/deploy/predict.py index 331246acc..464685331 100644 --- a/paddlespeech/cls/exps/panns/deploy/predict.py +++ b/paddlespeech/cls/exps/panns/deploy/predict.py @@ -59,7 +59,6 @@ def extract_features(files: str, **kwargs): feature_extractor = LogMelSpectrogram(sr, **kwargs) feat = feature_extractor(paddle.to_tensor(waveforms[i])) feat = paddle.transpose(feat, perm=[1, 0]) - feats.append(feat) return np.stack(feats, axis=0) @@ -78,6 +77,7 @@ class Predictor(object): if paddlespeech.utils.satisfy_paddle_version('3.0.0-beta'): config = inference.Config(model_dir, 'inference') + config.disable_mkldnn() else: model_file = os.path.join(model_dir, 'inference.pdmodel') params_file = os.path.join(model_dir, "inference.pdiparams") @@ -87,7 +87,6 @@ class Predictor(object): config = inference.Config(model_file, params_file) - config.disable_mkldnn() if device == "gpu": # set GPU configs accordingly # such as intialize the gpu memory, enable tensorrt