jingwenli0312 / Heterogeneous-Attentions-PDP-DRL

MIT License
45 stars 19 forks source link

Reimplementation in RL4CO 🚀 #4

Open fedebotu opened 1 year ago

fedebotu commented 1 year ago

Hi there 👋🏼

First of all, great job on your work, it has served as an inspiration to our research group! We are actively developing RL4CO, a library for all things Reinforcement Learning for Combinatorial Optimization. We reimplemented HAM in the library and modularized it to be compatible with other training schemes such as POMO, SymNCO and PPO. We also used some recent software (such as TorchRL, TensorDict, PyTorch Lightning, and Hydra). We tried to make everything as easy to use as possible in the hope of helping practitioners and researchers.

Feel free to check RL4CO out 🚀 ^ ^