avoid ".to(device)"

2024-09-28 07:58:15 +09:00
parent 669b7e1b43
commit 7433d1e877
5 changed files with 37 additions and 33 deletions
--- a/dreamer.py
+++ b/dreamer.py
@@ -258,8 +258,8 @@ def main(config):
        else:
            random_actor = torchd.independent.Independent(
                torchd.uniform.Uniform(
-                    torch.Tensor(acts.low).repeat(config.envs, 1),
-                    torch.Tensor(acts.high).repeat(config.envs, 1),
+                    torch.tensor(acts.low).repeat(config.envs, 1),
+                    torch.tensor(acts.high).repeat(config.envs, 1),
                ),
                1,
            )