Problem about the argument detection dataset

[njuwzt]

Dear authors:
I am glad to learn your work. I have a question about the argument detection dataset. In the file run_arg_detection.py, the training dataset(batch data) used in code is different from the dataset in the file run_trigger_detection.py. But I only find the file save_dataset.py which is used to save the trigger detection dataset. Could you tell me hou can I store the argument detection dataset into .pt file? Thank you for your work!

亲爱的作者：
您好！我很高兴能阅读您的论文。我注意到代码中论元抽取代码里导入的.pt文件应该与事件检测导入的.pt文件不同。但是save_dataset.py文件中只有保存事件检测数据集的代码，请问您能告诉我如何保存论元检测的数据到.pt文件中吗？