File tree Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Expand file tree Collapse file tree 2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -2824,7 +2824,7 @@ def preproc(data):
28242824 preproc ,
28252825 num_workers = max (1 , os .cpu_count () - 4 ),
28262826 num_chunks = 1000 ,
2827- mp_start_method = "fork" ,
2827+ # mp_start_method="fork",
28282828 pbar = True ,
28292829 dest = tmpdir ,
28302830 num_frames = 100 ,
@@ -3000,7 +3000,7 @@ def fn(data: TensorDict):
30003000 CloudpickleWrapper (fn ),
30013001 num_workers = max (1 , os .cpu_count () - 2 ),
30023002 num_chunks = 500 ,
3003- mp_start_method = "fork" ,
3003+ # mp_start_method="fork",
30043004 dest = tmpdir ,
30053005 )
30063006 dataset = ReplayBuffer (storage = new_storage )
Original file line number Diff line number Diff line change @@ -67,8 +67,8 @@ def _jumanji_to_torchrl_spec_transform(
6767 dtype = numpy_to_torch_dtype_dict [spec .dtype ]
6868 return BoundedTensorSpec (
6969 shape = shape ,
70- low = np .asarray (spec .low ),
71- high = np .asarray (spec .high ),
70+ low = np .asarray (spec .minimum ),
71+ high = np .asarray (spec .maximum ),
7272 dtype = dtype ,
7373 device = device ,
7474 )
You can’t perform that action at this time.
0 commit comments