Merge pull request #38 from lucasdelimanogueira/tmp

Fix long time step() detach
2024-05-07 01:35:29 -03:00 · 2024-05-07 01:35:29 -03:00 · 33481a1c07
commit 33481a1c07
parent a4dbfe8e7d ef3868066e
1 changed files with 0 additions and 1 deletions
--- a/norch/nn/module.py
+++ b/norch/nn/module.py
@ -52,7 +52,6 @@ class Module(ABC):

    def to(self, device):
        for _, _, parameter in self.parameters():
-            print(type(parameter))
            parameter.to(device)

        return self