Skip to content

Commit

Permalink
Merge branch 'master' of github.com:Flunzmas/gym-autokey
Browse files Browse the repository at this point in the history
  • Loading branch information
Flunzmas committed Nov 13, 2020
2 parents 776546e + fde1d38 commit 58a1b13
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,8 @@ The _TacticSelector_ defined in `scripts/tactic_selector.py` provides a class wr

## PPO-GoalRGCN

I am currently implementing a PPO agent that uses a relational GCN (see [the DGL library](https://docs.dgl.ai/index.html)). See `agents/goal_rgcn_agent.py` for the WIP code.
I have implemented a PPO agent that uses a relational GCN (see [the DGL library](https://docs.dgl.ai/index.html)). See `agents/goal_rgcn_agent.py` for the code.
It is able to train on simple training sets, although it needs **a lot** of RAM (>= 16GB for small Graphs). Future work might tackle this issue or test the performance on bigger machines than the ones I have access to.

# Attribution

Expand Down

0 comments on commit 58a1b13

Please sign in to comment.