UniRTL has been implemented and tested on PyTorch 2.7.1 with python 3.10.
We recommend you first use conda to create a virtual environment, and install pytorch following official installation instructions. For example:
conda create -n unirtl python=3.10
conda activate unirtl
pip install torch
You should also install torch_geometric and transformers
pip install torch_geometric
pip install transformers
Then you can train UniRTL with the scripts provided:
bash scripts/text_code_alignment.sh
bash scripts/graph_incorporation.sh