Skip to content

Commit 258befd

Browse files
committed
More updates
1 parent ed5b87b commit 258befd

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

intermediate_source/reinforcement_q_learning.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
77
88
This tutorial shows how to use PyTorch to train a Deep Q Learning (DQN) agent
9-
on the CartPole-v0 task from the `OpenAI Gym <https://www.gymlibrary.dev/>`__.
9+
on the CartPole-v1 task from the `OpenAI Gym <https://www.gymlibrary.dev/>`__.
1010
1111
**Task**
1212
@@ -74,7 +74,7 @@
7474
import torchvision.transforms as T
7575

7676

77-
env = gym.make('CartPole-v0', new_step_api=True, render_mode='single_rgb_array').unwrapped
77+
env = gym.make('CartPole-v1', new_step_api=True, render_mode='single_rgb_array').unwrapped
7878

7979
# set up matplotlib
8080
is_ipython = 'inline' in matplotlib.get_backend()

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ scikit-image
4343
scipy
4444
pillow==9.0.1
4545
wget
46-
gym==0.24.0
47-
gym-super-mario-bros==7.3.0
46+
gym==0.25.1
47+
gym-super-mario-bros==7.4.0
4848
timm
4949
pygame==2.1.2

0 commit comments

Comments
 (0)