made rnn input float 32

d3group · Aug 14, 2024 · 1209501 · 1209501
1 parent e7509da
commit 1209501
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ddopnew/RL_approximators.py b/ddopnew/RL_approximators.py
@@ -585,6 +585,6 @@ def forward(self, state, action):
             rnn_input = state
             mlp_input = action
 
-        a = self.model(rnn_input, mlp_input)
+        a = self.model(rnn_input.float(), mlp_input.float())
 
         return a
diff --git a/nbs/60_approximators/21_critic_networks.ipynb b/nbs/60_approximators/21_critic_networks.ipynb
@@ -696,7 +696,7 @@
     "            rnn_input = state\n",
     "            mlp_input = action\n",
     "\n",
-    "        a = self.model(rnn_input, mlp_input)\n",
+    "        a = self.model(rnn_input.float(), mlp_input.float())\n",
     "\n",
     "        return a"
    ]