Extending
One of the core tenets of BenchMARL is allowing users to leverage the existing algorithm and tasks implementations to benchmark their newly proposed solution.
For this reason we expose standard interfaces with simple abstract methods
for Algorithm
, Task
and Model
.
To introduce your solution in the library, you just need to implement the abstract methods exposed by these base classes which use objects from the TorchRL library.
Here is an example on how you can create a custom algorithm
Here is an example on how you can create a custom task
Here is an example on how you can create a custom model