beanie00 / Decision-ConvFormer

[ICLR 2024 Spotlight] Code for the paper "Decision ConvFormer: Local Filtering in MetaFormer is Sufficient for Decision Making"
https://www.beanie00.com/publications/dc
MIT License
7 stars 1 forks source link

Decision ConvFormer: Local Filtering in MetaFormer is Sufficient for Decision Making

Overview

Decision ConvFormer Architecture: image info

Acknowledgements

Our Decision ConvFormer code is based on decision-transformer.

Instructions

We provide code in two sub-directories: atari containing code for Atari experiments and gym containing code for OpenAI Gym experiments. See corresponding READMEs in each folder for instructions; scripts should be run from the respective directories. It may be necessary to add the respective directories to your PYTHONPATH.