jat-dataset-tokenized jat-project
Dataset Card for "jat-dataset-tokenized" More Information needed
- 種別
- dataset
- ダウンロード
- 662,576
- いいね
- 23
- アクセス
- public
- ファイル
- 0
タグ
- 強化学習
- 画像データセット
- マルチモーダル
- 事前学習コーパス
- テキストデータ
概要
本データセットは、多数のAtariゲーム環境（atari-alien、atari-breakout等）から収集された観測画像、報酬、離散アクション、attention_mask、loss_weightを含む、強化学習モデルのトレーニング用にトークン化済みの大規模データセットです。各ゲーム環境ごとにtrain/test分割が用意され、4次元の画像観察シーケンスを特徴として保有しており、Atari上で動作するRLエージェントや世界モデルの学習・評価に適しています。JAT（Jointly-trained Agent Transformer）プロジェクトの研究用途に活用されることが想定されます。
README
--- dataset_info: - config_name: atari-alien features: - name: image_observations sequence: sequence: sequence: sequence: float32 - name: rewards sequence: float32 - name: discrete_actions sequence: int64 - name: attention_mask sequence: int8 - name: loss_weight sequence: float32 splits: - name: tr…