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

How to use the checkpointer to save replay buffer in the disk? #12

Closed fmxFranky closed 3 years ago

fmxFranky commented 4 years ago

I found that there are no files in the directory of the checkpointer's folder, neigther /tmp/xxxxx or the specific predefined directory. I also noticed that the TFCheckpointer class have inferfaces for saving and loading replay buffer, but when i call replay_checkpointer.save() i got the error said "Don't have the attribute 'save' ". So how to use the checkpointer class or reverb's saver class to save replay buffer in the disk? Thanks~

jmacglashan commented 4 years ago

Use client.checkpoint()

Then when you restore, point to the parent directory of the returned directory (using the specific checkpoint doesn't seem to work right now)

ebrevdo commented 3 years ago

(James gave the solution)