diff --git a/torchrl/_utils.py b/torchrl/_utils.py index 9f04d3f87c1..7de6453d10a 100644 --- 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