Skip to content

Implementation of Multi-Game Decision Transformers in PyTorch

License

Notifications You must be signed in to change notification settings

etaoxing/multigame-dt

Repository files navigation

Implementation of Multi-Game Decision Transformers in PyTorch

Quickstart

conda create --name mgdt python=3.10
conda activate mgdt
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html
pip install -r requirements.txt

python scripts/download_weights.py
python run.py

Baselines

logs

model params task this repo orig.
mgdt 200M Breakout 298.8 290.6

References:

Releases

No releases published

Packages

No packages published

Languages