ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
https://rl4.co
MIT License
455 stars 84 forks source link

[Feat] Nonautoregressive Methods #122

Closed cbhua closed 8 months ago

cbhua commented 9 months ago

Description

Contributed by @Furffico, @tycbony, and @henry-yeh from DeepACO and GLOP. Adding new support for non-autoregressive (NAR) methods, including edge embedding methods for various environments, anisotropic GNN encoder with edge-gating mechanism, non-autoregressive decoder, and related utils functions.

Motivation and Context

NAR methods are an important extension for RL4CO model support, aside from various previous autoregressive methods.

Types of changes

Checklist

Furffico commented 9 months ago

I have modified the code following the review above. Please review the changes.