Skip to content

Commit 6a42116

Browse files
author
Vincent Moens
authored
[BugFix] Temporarily set lazy legacy to True (#1840)
1 parent 6f90397 commit 6a42116

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

torchrl/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,12 @@
77

88
import torch
99

10+
from tensordict import set_lazy_legacy
11+
1012
from torch import multiprocessing as mp
1113

14+
set_lazy_legacy(True).set()
15+
1216
if torch.cuda.device_count() > 1:
1317
n = torch.cuda.device_count() - 1
1418
os.environ["MUJOCO_EGL_DEVICE_ID"] = str(1 + (os.getpid() % n))

0 commit comments

Comments
 (0)