Our experiments are conducted on the following evironmental settings. To ensure reproductivity, we strongly recommand that you run our code on the same settings.
- GPU: NVIDIA RTX 3090 24G
- CUDA Version: 11.4
- Pytorch Version: 1.10.0
- Python Version: 3.9.7
Training:
python main.py
All the parameters are listed in main.py
You can download datasets from link in our paper. The code of preprocessing will be published soon.