TensorStorageCheckpointer

class torchrl.data.replay_buffers.TensorStorageCheckpointer

TensorStorage 的儲存檢查點。

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

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