Code for "Enhancing Inter-Sentence Attention for Semantic Textual Similarity"
We used python=3.8
, cudatoolkit=11.1
, transformers==4.11.3
, torch==1.11.0
Data Preprocessing:
python pre-dataset.py
Train and test the model:
bash run.sh
Please download the data here, and put it in 'datasets/' folder.