Only initialize DDPG target when training is not resumed

3366990b · Nicola Gatto · 4ed3b332 · 3366990b
Commit 3366990b authored Jun 11, 2019 by Nicola Gatto
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl ...n/resources/templates/gluon/reinforcement/agent/Agent.ftl +3 -3

No files found.
--- a/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
+++ b/src/main/resources/templates/gluon/reinforcement/agent/Agent.ftl
@@ -465,9 +465,9 @@ class DdpgAgent(Agent):
        else:
            self._training_stats = DdpgTrainingStats(episodes)

-        # Initialize target Q' and mu'
-        self._actor_target = self._copy_actor()
-        self._critic_target = self._copy_critic()
+            # Initialize target Q' and mu'
+            self._actor_target = self._copy_actor()
+            self._critic_target = self._copy_critic()

        # Initialize l2 loss for critic network
        l2_loss = gluon.loss.L2Loss()