Update README.md

README.md

@@ -2,21 +2,26 @@
 library_name: transformers
 tags:
 - generated_from_trainer
+- decision_transformer
 model-index:
-- name: output
+- name: decision_transformer_gym_hopper
   results: []
+datasets:
+- edbeeching/decision_transformer_gym_replay
+base_model:
+- edbeeching/decision-transformer-gym-hopper-medium
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# output
+# decision_transformer_gym_hopper
 
-This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
+This model is a fine-tuned version of [](https://huggingface.co/) on edbeeching/decision_transformer_gym_replay.
 
 ## Model description
 
-More information needed
+Decision Transformer training on gym hoppers (RL dataset).
 
 ## Intended uses & limitations
 
@@ -49,4 +54,4 @@ The following hyperparameters were used during training:
 - Transformers 4.44.2
 - Pytorch 2.4.1+cu121
 - Datasets 3.0.1
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1