Update dqn_with_rnn_tutorial.py

sekyondaMeta · web-flow · commit d18ba1197b53 · 2025-07-14T20:10:20.000-04:00
https://github.com/pytorch/tutorials/actions/runs/16279151183/job/45965049712?pr=3463#step:9:8173
diff --git a/intermediate_source/dqn_with_rnn_tutorial.py b/intermediate_source/dqn_with_rnn_tutorial.py
@@ -342,7 +342,8 @@
 # will return a new instance of the LSTM (with shared weights) that will
 # assume that the input data is sequential in nature.
 #
-policy = Seq(feature, lstm.set_recurrent_mode(True), mlp, qval)
+lstm = torch.nn.LSTM(input_size, hidden_size, num_layers, default_recurrent_mode=True)
+policy = Seq(feature, lstm, mlp, qval)
 
 ######################################################################
 # Because we still have a couple of uninitialized parameters we should

Original file line number	Diff line number	Diff line change
`@@ -342,7 +342,8 @@`
`342`	`342`	`# will return a new instance of the LSTM (with shared weights) that will`
`343`	`343`	`# assume that the input data is sequential in nature.`
`344`	`344`	`#`
`345`		`-policy = Seq(feature, lstm.set_recurrent_mode(True), mlp, qval)`
	`345`	`+lstm = torch.nn.LSTM(input_size, hidden_size, num_layers, default_recurrent_mode=True)`
	`346`	`+policy = Seq(feature, lstm, mlp, qval)`
`346`	`347`
`347`	`348`	`######################################################################`
`348`	`349`	`# Because we still have a couple of uninitialized parameters we should`