search

nsidn98 / InforMARL

Code for our paper: Scalable Multi-Agent Reinforcement Learning through Intelligent Information Aggregation
https://nsidn98.github.io/InforMARL/
MIT License
91 stars 22 forks source link

Hi, could you please tell how to use UniMP in your InforMARL? I can neither find it in your paper nor in your code. Thx. #5

Closed TianyQi closed 11 months ago

TianyQi commented 11 months ago

Hi, could you please tell how to use UniMP in your InforMARL? I can neither find it in your paper nor in your code. Thx.

nsidn98 commented 11 months ago

Hi @TianyQi, we use the graph transformer operator defined in UniMP here. You can find the PyTorch Geometric documentation for that layer here.

TianyQi commented 11 months ago

Thanks a lot for your reply~

nsidn98 commented 11 months ago

Closing this comment because the issue seems to be resolved. Reopen if required.