TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

用於 TensorStorages 的儲存檢查點器。

此類支援基於 TensorDict 的儲存以及 pytree。

如果提供了儲存和載入鉤子，此類將呼叫它們。這些鉤子應將要轉換的資料以及應儲存資料的路徑作為輸入。