File tree 2 files changed +4
-4
lines changed
2 files changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -2824,7 +2824,7 @@ def preproc(data):
2824
2824
preproc ,
2825
2825
num_workers = max (1 , os .cpu_count () - 4 ),
2826
2826
num_chunks = 1000 ,
2827
- mp_start_method = "fork" ,
2827
+ # mp_start_method="fork",
2828
2828
pbar = True ,
2829
2829
dest = tmpdir ,
2830
2830
num_frames = 100 ,
@@ -3000,7 +3000,7 @@ def fn(data: TensorDict):
3000
3000
CloudpickleWrapper (fn ),
3001
3001
num_workers = max (1 , os .cpu_count () - 2 ),
3002
3002
num_chunks = 500 ,
3003
- mp_start_method = "fork" ,
3003
+ # mp_start_method="fork",
3004
3004
dest = tmpdir ,
3005
3005
)
3006
3006
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(
67
67
dtype = numpy_to_torch_dtype_dict [spec .dtype ]
68
68
return BoundedTensorSpec (
69
69
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 ),
72
72
dtype = dtype ,
73
73
device = device ,
74
74
)
You can’t perform that action at this time.
0 commit comments