Skip to content

Commit caa0094

Browse files
authored
Update dqn_with_rnn_tutorial.py
Resetting changes in favor of: #3462
1 parent c391f15 commit caa0094

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

intermediate_source/dqn_with_rnn_tutorial.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,6 @@
238238
device=device,
239239
in_key="embed",
240240
out_key="embed",
241-
default_recurrent_mode=True,
242241
)
243242

244243
######################################################################
@@ -343,7 +342,7 @@
343342
# will return a new instance of the LSTM (with shared weights) that will
344343
# assume that the input data is sequential in nature.
345344
#
346-
policy = Seq(feature, lstm, mlp, qval)
345+
policy = Seq(feature, lstm.set_recurrent_mode(True), mlp, qval)
347346

348347
######################################################################
349348
# Because we still have a couple of uninitialized parameters we should

0 commit comments

Comments
 (0)