From 6a98a26de6219bfe33f40ee5ac60ec13a6c27784 Mon Sep 17 00:00:00 2001 From: "D. Debnath" Date: Wed, 15 Mar 2023 16:04:09 +0530 Subject: [PATCH] Update vector_envs_tutorial.py (#390) --- docs/tutorials/gymnasium_basics/vector_envs_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/tutorials/gymnasium_basics/vector_envs_tutorial.py b/docs/tutorials/gymnasium_basics/vector_envs_tutorial.py index 90fd93715..2e2bb06e4 100644 --- a/docs/tutorials/gymnasium_basics/vector_envs_tutorial.py +++ b/docs/tutorials/gymnasium_basics/vector_envs_tutorial.py @@ -454,7 +454,7 @@ for sample_phase in tqdm(range(n_updates)): # perform the action A_{t} in the environment to get S_{t+1} and R_{t+1} states, rewards, terminated, truncated, infos = envs_wrapper.step( - actions.numpy() + actions.cpu().numpy() ) ep_value_preds[step] = torch.squeeze(state_value_preds)