From 4452d636ca03cc6597fecc9ff6e4dd4aa2b4dbee Mon Sep 17 00:00:00 2001 From: Nagaloop Date: Mon, 21 Mar 2022 09:33:24 +0000 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=94=B9import=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py b/ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py index 637e015cc9..314d0090cf 100644 --- a/ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py +++ b/ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py @@ -16,9 +16,9 @@ import sys import argparse import pickle as pkl import torch -from models import TextCNN sys.path.append(r'./Chinese-Text-Classification-Pytorch') +from models import TextCNN parser = argparse.ArgumentParser(description='TextCNN_pth2onnx.py') parser.add_argument('--weight_path', required=True, help='Path to model weight file, abs path recommended.') -- Gitee