Closed RoyalSkye closed 5 months ago
I have added an example to 1-mtvrp.ipynb
. It seems that the notebook has some issues on my laptop, so I just uploaded a raw version without output ๐.
It seems that the notebook has some issues on my laptop, so I just uploaded a raw version without output
Are you using an M-series Macbook? We had some problems only with MTVRP and only with Apple MPS accelerators :rofl:
The notebook works smoothly for me! If it has some bug on a Macbook, it's only could be the problem with the M chip ๐ We had experiences on this.
Maybe this is the reason๐. Could you help me to upload a new version that overwrites the current one? Thanks!
Thank Chuanbo. Looks great! I have nothing to add to this PR.
Great job! After @LTluttmann having a look let's launch! ๐
awesome work! Clean and well documented code, was a pleasure to read it haha! Merging now
Description
This PR adds support for
MVMoE
in the multi-task VRP setting. In specific, the token-level input-choice gating is implemented. Some minor bugs are fixed as well, such as missing the definition ofMTVRPInitEmbedding
andMTVRPContext
.1-mtvrp.ipynb
.Motivation and Context
A step towards foundation VRP models.
Types of changes
What types of changes does your code introduce? Remove all that do not apply:
Checklist
Go over all the following points, and put an
x
in all the boxes that apply. If you are unsure about any of these, don't hesitate to ask. We are here to help!