[Refactor] change default CKPT_BACKEND to torch (#1830)

pytorch · Jan 23, 2024 · 0699753 · 0699753
1 parent 5b67dd3
commit 0699753
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchrl/_utils.py b/torchrl/_utils.py
@@ -178,7 +178,7 @@ class _Dynamic_CKPT_BACKEND:
     backends = ["torch", "torchsnapshot"]
 
     def _get_backend(self):
-        backend = os.environ.get("CKPT_BACKEND", "torchsnapshot")
+        backend = os.environ.get("CKPT_BACKEND", "torch")
         if backend == "torchsnapshot":
             try:
                 import torchsnapshot  # noqa: F401