Fix long time step() detach

This commit is contained in:
lucasdelimanogueira 2024-05-07 01:30:45 -03:00
parent 9634ab12c5
commit ef3868066e

View file

@ -52,7 +52,6 @@ class Module(ABC):
def to(self, device):
for _, _, parameter in self.parameters():
print(type(parameter))
parameter.to(device)
return self