Skip to content

Latest commit

 

History

44 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Transformers applied to RL tasks

Transformers trained with reinforce and V-MPO

VMPO

GTrXL transformer

architecture Run tests/gtrxl_test.py with --state_rep=gtrxl

CoBERL

Run tests/gtrxl_test.py with --state_rep=coberl to run the architecture with CoBERL in the place of GTrXL in architecture

Reinforce

To run the GTrXL as a policy-estimating network, run pg.py with bool Trans=True within. To evaluate the common MLP estimation, set this bool Trans=False.

About

Transformers (GTrXL & CoBERL) applied to RL tasks

Resources

Stars

29 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages