search

nicklashansen / tdmpc2

Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"
https://www.tdmpc2.com
MIT License
343 stars 71 forks source link

cannot conduct "mt" experiments #35

Closed kaiyama12345679 closed 3 months ago

kaiyama12345679 commented 3 months ago

When I try

python train.py task=mt80 model_size=48 batch_size=1024

I got the following error

AssertionError: Expected data directory /root/data to contain mt80, please double-check your config
nicklashansen commented 3 months ago

Hi @kaiyama12345679, did you download the mt80 dataset from our website? It is available here: https://www.tdmpc2.com/dataset After downloading, you will need to point to the dataset using the data_dir argument. I hope this helps!

kaiyama12345679 commented 3 months ago

@nicklashansen, thank you so much for your reply. I have downloaded the mt30 and mt80 from your huggingface repository to /root/data, and set data_dir=/root/data. However the above error still occured in my environments.

kaiyama12345679 commented 3 months ago

I'm sorry, I misunderstood. When I set data_dir=/root/data/mt30 or data_dir=/root/data/mt80, it works! Thank you for your help!

nicklashansen commented 3 months ago

Great that you solved it!