- Python 3.8.10
- PyTorch 1.21.1
- transformers 4.10.0
You need
python preprocess_graph_c.py
python train_and_extract_graph_features.py
python train_and_extract_graph_features_c.py
python run_roberta_rel.py (adding --do_train to the args)
Trained model can be downloaded from : https://drive.google.com/file/d/11YSO_BOpYCDR08FyxjpX3xi7M1O2LmRK/view?usp=sharing
Dataset can be downloaded from : https://scale.com/open-av-datasets/oxford