The official code repository for CaDA: Cross-Problem Routing Solver with Constraint-Aware Dual-Attention.
- Clone the repository:
git clone https://github.com/CIAM-Group/CaDA.git- Download datasets:
- Download
data.zipfrom Hugging Face. - Unzip
data.zipand organize the files in the project directory as follows:
CaDA
├── data
│ ├── lib_data
│ └── synthetic_data
├── 50
├── 100
└── utils
- Download checkpoints:
- Create 'result' folder manually under 'CaDA/50' and 'CaDA/100'.
- Download
checkpoint.zipfrom Hugging Face. - Unzip
checkpoint.zip. It will produce two directories:50and100.- Inside
50, you will find a folder named2024-1111-1139. - Inside
100, you will find a folder named2024-1121-1355.
- Inside
- Organize them into the project directory as follows:
CaDA
├── data
│ ├── lib_data
│ └── synthetic_data
├── 50
│ └── result
│ └── 2024-1111-1139
├── 100
│ └── result
│ └── 2024-1121-1355
└── utils
- Prepare environment:
The project is developed with Python 3.8.15. Key packages include:
torch 2.0.1
torchrl 0.1.1
rl4co 0.2.0
tensordict 0.1.2
The complete list of dependencies can be found in requirements.txt.
For detailed instructions on training and testing the model, please refer to the README files inside the 50 and 100 directories.