facebookresearch / BenchMARL

A collection of MARL benchmarks based on TorchRL
https://benchmarl.readthedocs.io/
MIT License
216 stars 27 forks source link

[Feature] GNN model #18

Closed matteobettini closed 10 months ago

matteobettini commented 11 months ago

This PR aims at introducing a GNN model customizable via torch_geometric.

This is currently on hold because of:

https://github.com/pytorch/rl/issues/1576