takuseno / d4rl-atari

Datasets for data-driven deep reinforcement learning with Atari (wrapper for datasets released by Google)
MIT License
101 stars 14 forks source link

Loading datasets #14

Closed mamengyiyi closed 1 year ago

mamengyiyi commented 1 year ago

Hi, I found that even I have downloaded all the datasets of 50 epochs, d3rlpy still only loads the data of index 1 and epoch 1. How can I make it load all the data of 50 epochs? Or how can I specify the data it loads?

mamengyiyi commented 1 year ago

Hi, I found that even I have downloaded all the datasets of 50 epochs, d3rlpy still only loads the data of index 1 and epoch 1. How can I make it load all the data of 50 epochs? Or how can I specify the data it loads?

Here is the log: image

Below is my datasets directory: image

mamengyiyi commented 1 year ago

Oh, when I specify the dataset argument as breakout-expert-v0, it loads the data of epoch 50, and when I specify it as breakout-mixed-v0, it loads the data of epoch 1.