Skip to content

README.md

Latest commit

 

History

History
10 lines (10 loc) · 526 Bytes

README.md

File metadata and controls

10 lines (10 loc) · 526 Bytes

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.