removed scheduling function

2023-09-26 20:58:55 +09:00
parent 2cdba230d8
commit 16635df3e4
4 changed files with 17 additions and 54 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -40,16 +40,6 @@ class Dreamer(nn.Module):
        # this is update step
        self._step = logger.step // config.action_repeat
        self._update_count = 0
-        # Schedules.
-        config.actor_entropy = lambda x=config.actor_entropy: tools.schedule(
-            x, self._step
-        )
-        config.actor_state_entropy = (
-            lambda x=config.actor_state_entropy: tools.schedule(x, self._step)
-        )
-        config.imag_gradient_mix = lambda x=config.imag_gradient_mix: tools.schedule(
-            x, self._step
-        )
        self._dataset = dataset
        self._wm = models.WorldModel(obs_space, act_space, self._step, config)
        self._task_behavior = models.ImagBehavior(