- LwD@Learning What to Defer for Maximum Independent Sets
- DGL-TreeSearch@WHAT’S WRONG WITH DEEP LEARNING IN TREE SEARCH FOR COMBINATORIAL OPTIMIZATION
sudo docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -v ./:/L2M/ -it -p 8888:8888 nvcr.io/nvidia/pytorch:22.07-py3
pip install dgl -f https://data.dgl.ai/wheels/cu118/repo.html
pip install dglgo -f https://data.dgl.ai/wheels-test/repo.html
sudo docker run --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864 --rm -v ./:/L2M/ -it -p 8888:8888 nvcr.io/nvidia/pytorch:22.07-py3
cd /L2M
For training,
python run_l2m.py --mode train
For testing,
python run_l2m.py --mode test