Edward Beeching
added checkpoints and readme file
7978894
raw
history blame
314 Bytes
name: decision-transformer-atari
channels:
- pytorch
dependencies:
- python=3.7.9
- pytorch
- cudatoolkit=11.3
- numpy
- psutil
- opencv
- pip
- pip:
- atari-py
- pyprind
- tensorflow-gpu>=1.13
- absl-py
- atari-py
- gin-config
- gym
- tqdm
- blosc
- git+https://github.com/google/dopamine.git