ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

Home Page:https://rl4.co

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Refactoring][Minor] split the decoder and the policy for the mdam model

cbhua opened this issue · comments

The first submitted version of the MDAM model works properly. But the code is massive, where the decoder process is included in the policy. Do a refactoring to split the decoder and policy to make the structure clear and meet our flexible repository requirement.

This task will not be long, I will use one or two hours today (Saturday) to finish this.

Done; now the MDAM model has encoder.py, decoder.py, and the policy.py.