本回答引用这个回答
# coding: UTF-8
import time, os
import numpy as np
#from train_eval import train, test
import random
#from bert import Model, Config
import argparse
#from utils import build_dataset, build_iterator, get_time_dif, load_dataset, gettoken
from torch.utils.data import TensorDataset, DataLoader
import torch
import json
'''
import sys
sys.argv=['']
del sys
'''
parser = argparse.ArgumentParser(description='Salient triple classification')
parser.add_argument("--do_train", type=bool, default=True, help="Whether to run training.",)
parser.add_argument("--test_batch", default=200, type=int, help="Test every X updates steps.")
parser.add_argument("--data_dir", default="/home/daidehao/code/OpenBG-CSK/data", type=str, help="The task data directory.")
parser.add_argument("--model_dir", default="bert-base-chinese", type=str, help="The directo