[BugFix] revert MHA to Kool's implementation

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

Closed fedebotu closed 1 year ago