kzl / decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

MIT License

2.33k stars 440 forks source link

Closed wilhem closed 5 months ago

wilhem commented 5 months ago

I downloaded the repository but at the line

load dataset

dataset_path = f"data/{env_name}-{dataset}-v2.pkl" with open(dataset_path, "rb") as f: trajectories = pickle.load(f)

it quits with an error, since that pickle dataset is not found.

In the repository there is no dataset. How did you get it?

Thanks

wilhem commented 5 months ago

Ok, sorry I found them. I just read the wrong readme