Commit a42a43f9 authored by Nicola Gatto's avatar Nicola Gatto

Adjust test to gluon templates

parent 332c5db2
Pipeline #149949 failed with stage
in 4 minutes and 54 seconds
......@@ -906,4 +906,4 @@ class DqnAgent(Agent):
def _save_current_as_best_net(self):
self._best_net = copy_net(
self._qnet, (1,) + self._state_dim, ctx=self._ctx)
self._qnet, self._state_dim, ctx=self._ctx)
......@@ -906,4 +906,4 @@ class DqnAgent(Agent):
def _save_current_as_best_net(self):
self._best_net = copy_net(
self._qnet, (1,) + self._state_dim, ctx=self._ctx)
self._qnet, self._state_dim, ctx=self._ctx)
......@@ -906,4 +906,4 @@ class DqnAgent(Agent):
def _save_current_as_best_net(self):
self._best_net = copy_net(
self._qnet, (1,) + self._state_dim, ctx=self._ctx)
self._qnet, self._state_dim, ctx=self._ctx)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment