edbeeching/decision_transformer_gym_replay
Viewer • Updated • 10.7k • 701 • 5
How to use hishamcse/decision-transformer-halfcheetah-v4 with Transformers:
# Load model directly
from transformers import AutoTokenizer, TrainableDT
tokenizer = AutoTokenizer.from_pretrained("hishamcse/decision-transformer-halfcheetah-v4")
model = TrainableDT.from_pretrained("hishamcse/decision-transformer-halfcheetah-v4", device_map="auto")This model was trained from scratch on the halfcheetah-expert-v2 dataset. For full code see: https://www.kaggle.com/code/syedjarullahhisham/drl-adv-hf-decisiontransformer-halfcheetah-hopper
Github repos(Give a star if found useful):
Kaggle Notebook:
The following hyperparameters were used during training: