diff --git a/ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py b/ACL_PyTorch/built-in/nlp/textcnn/TextCNN_pth2onnx.py index 637e015cc9a00ea543dc5605a43f10d6f39e31ea..314d0090cf19dc9c1cdfbf0ae4bb11e0331cb27b 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.')