Downloads · 30 days
6
5% of all-time downloads
Decision_Transformer_HalfCheetahBullet_v4
MattStammers/Decision_Transformer_HalfCheetahBullet_v4
Decision_Transformer_HalfCheetahBullet_v4 is a machine learning model from MattStammers. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
should probably proofread and complete it, then remove this comment. --
All-time downloads
116
Public
Repo size
8.8 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin5 MB · 58%
At a glance
- Library
- transformers
- Model type
- decision_transformer
- Access
- Public
- Created
- Sep 18, 2023
- Updated
- Sep 18, 2023
- SHA
- 044ba415
Datasets
Model card
From the Hugging Face model README
<!-- 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. -->
Decision Transformer
This model is a fine-tuned version of on the decision_transformer_gym_replay dataset.
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.0001
- train_batch_size: 64
- eval_batch_size: 8
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- lr_scheduler_warmup_ratio: 0.1
- num_epochs: 120
Training results
As you can see the half-cheetah is ridiculously fast!
Framework versions
- Transformers 4.33.2
- Pytorch 2.0.1+cu118
- Datasets 2.14.5
- Tokenizers 0.13.3
Playing the Video
The video upload preview is not working yet on this. The cheetah runs off the screen at the end and is extremely fast and effective. If you download the .mp4 you can run it in your local machine
- Library
- transformers
- Type
- decision_transformer
- Created
- Sep 18, 2023
- Updated
- Sep 18, 2023