Update dqn_with_rnn_tutorial.py

sekyondaMeta · web-flow · commit c0d83873e1a7 · 2025-07-15T07:31:40.000-04:00
diff --git a/intermediate_source/dqn_with_rnn_tutorial.py b/intermediate_source/dqn_with_rnn_tutorial.py
@@ -238,6 +238,7 @@
     device=device,
     in_key="embed",
     out_key="embed",
+    default_recurrent_mode=True,
 )
 
 ######################################################################
@@ -342,7 +343,6 @@
 # will return a new instance of the LSTM (with shared weights) that will
 # assume that the input data is sequential in nature.
 #
-lstm = torch.nn.LSTM(input_size, hidden_size, num_layers, default_recurrent_mode=True)
 policy = Seq(feature, lstm, mlp, qval)
 
 ######################################################################

Original file line number	Diff line number	Diff line change
`@@ -238,6 +238,7 @@`
`238`	`238`	`device=device,`
`239`	`239`	`in_key="embed",`
`240`	`240`	`out_key="embed",`
	`241`	`+ default_recurrent_mode=True,`
`241`	`242`	`)`
`242`	`243`
`243`	`244`	`######################################################################`
`@@ -342,7 +343,6 @@`
`342`	`343`	`# will return a new instance of the LSTM (with shared weights) that will`
`343`	`344`	`# assume that the input data is sequential in nature.`
`344`	`345`	`#`
`345`		`-lstm = torch.nn.LSTM(input_size, hidden_size, num_layers, default_recurrent_mode=True)`
`346`	`346`	`policy = Seq(feature, lstm, mlp, qval)`
`347`	`347`
`348`	`348`	`######################################################################`