search

google-deepmind / reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
Apache License 2.0
704 stars 92 forks source link

Pb using TrajectoryDataset #83

Closed mikygit closed 3 years ago

mikygit commented 3 years ago

Hello, Is there a way to use a TrajectoryDataset without having to specify tables signatures? Or use some kinda 'Any' keyword?

Would be quiet usefull to use the replay buffer on different kind of gym environement for instance.

Thanx!

nb: using pytorch

acassirer commented 3 years ago

Hi,

I'm afraid that will not be possible since tf.data requires that the shapes and dtypes are known.