Skip to content

Commit 0699753

Browse files
author
Vincent Moens
authored
[Refactor] change default CKPT_BACKEND to torch (#1830)
1 parent 5b67dd3 commit 0699753

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrl/_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ class _Dynamic_CKPT_BACKEND:
178178
backends = ["torch", "torchsnapshot"]
179179

180180
def _get_backend(self):
181-
backend = os.environ.get("CKPT_BACKEND", "torchsnapshot")
181+
backend = os.environ.get("CKPT_BACKEND", "torch")
182182
if backend == "torchsnapshot":
183183
try:
184184
import torchsnapshot # noqa: F401

0 commit comments

Comments
 (0)