decisionTransformer / README.md
ciCic's picture
Update README.md
b51df79
---
tags:
- decisionTransformer
- deep reinforcement
datasets:
- edbeeching/decision_transformer_gym_replay
license:
- mit
---
### Running training
- Num examples = 1000
- Num Epochs = 120
- Instantaneous batch size per device = 64
- Total train batch size = 64
- Gradient Accumulation steps = 1
- Total optimization steps = 1920
### Train Output
- global_step = 1920
- train_runtime = 1849.2158
- train_samples_per_second = 64.892
- train_steps_per_second = 1.038
- train_loss = 0.04717305501302083
- epoch = 120.0
### Dataset
- edbeeching/decision_transformer_gym_replay
- halfcheetah-expert-v2