This repository contains the implementation code for our research paper, "Ant Colony Sampling with GFlowNets for Combinatorial Optimization." We have developed our codes based on the DeepACO codebase.
- Python 3.11.5
- PyTorch 2.1.1
- PyTorch Geometric 2.4.0
For the complete list of dependencies, please refer to the requirements.txt
file.
For the usage of the code, please refer to each folder's README.md
file.
If you find our work useful in your research, please consider citing our paper:
@article{,
title={Ant Colony Sampling with GFlowNets for Combinatorial Optimization},
author={Minsu Kim and Sanghyeok Choi and Jiwoo Son and Hyeonah Kim and Jinkyoo Park and Yoshua Bengio},
journal={arXiv preprint arXiv:2403.07041},
year={2024},
}