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/networks.py
+++ b/networks.py
@@ -215,7 +215,9 @@ class RSSM(nn.Module):
                    is_first,
                    is_first.shape + (1,) * (len(val.shape) - len(is_first.shape)),
                )
-                prev_state[key] = val * (1.0 - is_first_r) + init_state[key] * is_first_r
+                prev_state[key] = (
+                    val * (1.0 - is_first_r) + init_state[key] * is_first_r
+                )

        prior = self.img_step(prev_state, prev_action, None, sample)
        if self._shared: