From 37c45ee8d7e4db18d95677c203b3432f3e6dde80 Mon Sep 17 00:00:00 2001 From: nichongjia-2007 <nichongjia@gmail.com> Date: 星期四, 23 三月 2023 18:33:21 +0800 Subject: [PATCH] add batch interval for saving model --- funasr/export/test_torchscripts.py | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/funasr/export/test_torchscripts.py b/funasr/export/test_torchscripts.py index 11be763..9afec74 100644 --- a/funasr/export/test_torchscripts.py +++ b/funasr/export/test_torchscripts.py @@ -2,7 +2,7 @@ import numpy as np if __name__ == '__main__': - onnx_path = "/mnt/workspace/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/model.torchscripts" + onnx_path = "/nfs/zhifu.gzf/export/damo/speech_paraformer-large_asr_nat-zh-cn-16k-common-vocab8404-pytorch/model.torchscripts" loaded = torch.jit.load(onnx_path) x = torch.rand([2, 21, 560]) -- Gitblit v1.9.1