gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-07-31 05:44:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update vector_envs_tutorial.py (#390)

Browse Source
This commit is contained in:
D. Debnath
2023-03-15 16:04:09 +05:30
committed by GitHub
parent 745375f6e5
commit 6a98a26de6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/tutorials/gymnasium_basics/vector_envs_tutorial.py
View File

@@ -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)