erased unnecessary lines

2023-06-17 15:27:09 +09:00
parent 6c861ca7cb
commit f7c505579c
4 changed files with 12 additions and 8 deletions
--- a/models.py
+++ b/models.py
@@ -399,9 +399,6 @@ class ImagBehavior(nn.Module):
        if self._config.future_entropy and self._config.actor_state_entropy() > 0:
            reward += self._config.actor_state_entropy() * state_ent
        value = self.value(imag_feat).mode()
-        # value(15, 960, ch)
-        # action(15, 960, ch)
-        # discount(15, 960, ch)
        target = tools.lambda_return(
            reward[:-1],
            value[:-1],